結果

問題 No.1241 Eternal Tours
ユーザー hitonanode
提出日時 2020-09-07 00:56:43
言語 Java
(openjdk 23)
結果
AC  
実行時間 1,144 ms / 6,000 ms
コード長 4,219 bytes
コンパイル時間 2,654 ms
コンパイル使用メモリ 77,264 KB
実行使用メモリ 90,124 KB
最終ジャッジ日時 2024-11-29 11:09:46
合計ジャッジ時間 27,605 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

public class Main {
    final static long md = 998244353;
    final static long root = 3;

    static void ntt(int n, long[] a, long[] w)
    {
        if (n == 1) return;
        for (int m = n / 2; m > 0; m /= 2)
        {
            for (int s = 0, k = 0; s < n; s += 2 * m, k++)
            {
                for (int i = s; i < s + m; i++)
                {
                    long x = a[i], y = a[i + m] * w[k] % md;
                    a[i] = x + y;
                    if (a[i] >= md)
                    {
                        a[i] -= md;
                    }
                    a[i + m] = x - y + md;
                    if (a[i + m] >= md)
                    {
                        a[i + m] -= md;
                    }
                }
            }
        }
    }
    static void intt(int n, long[] a, long[] iw)
    {
        if (n == 1) return;
        for (int m = 1; m < n; m *= 2)
        {
            for (int s = 0, k = 0; s < n; s += 2 * m, k++)
            {
                for (int i = s; i < s + m; i++)
                {
                    long x = a[i], y = a[i + m];
                    a[i] = x + y;
                    if (a[i] >= md)
                    {
                        a[i] -= md;
                    }
                    a[i + m] = (x - y + md) * iw[k] % md;
                }
            }
        }
        long n_inv = power(n, -1);
        for (int i = 0; i < n; i++)
        {
            a[i] = a[i] * n_inv % md;
        }
    }

    static long power(long x, long n)
    {
        if (n == -1)
        {
            n = md - 2;
        }
        x %= md;
        long ans = 1;
        long tmp = x;
        while (n > 0)
        {
            if ((n & 1) == 1)
            {
                ans = ans * tmp % md;
            }
            tmp = tmp * tmp % md;
            n /= 2;
        }
        return ans;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        final int X = sc.nextInt();
        final int Y = sc.nextInt();
        final long T = sc.nextLong();
        final int a = sc.nextInt();
        final int b = sc.nextInt();
        final int c = sc.nextInt();
        final int d = sc.nextInt();

        final int H = 1 << (X + 1);
        final int W = 1 << (Y + 1);

        long dp[][] = new long[H][W];
        long trans[][] = new long[H][W];

        dp[a][b] = 1;
        dp[a][W - b] = -1;
        dp[H - a][b] = -1;
        dp[H - a][W - b] = 1;
        trans[0][0] = trans[0][1] = trans[1][0] = trans[H - 1][0] = trans[0][W - 1] = 1;

        long w[] = new long[1 << 17];
        long iw[] = new long[1 << 17];
        w[0] = iw[0] = 1;
        for (int m = 1; m < 1 << 17; m *= 2)
        {
            long dw = power(root, (md - 1) / (4 * m));
            long dwinv = power(dw, -1);
            for (int i = 0; i < m; i++)
            {
                w[m + i] = w[i] * dw % md;
                iw[m + i] = iw[i] * dwinv % md;
            }
        }

        for (int i = 0; i < H; i++)
        {
            ntt(W, dp[i], w);
            ntt(W, trans[i], w);
        }
        for (int j = 0; j < W; j++)
        {
            long dpv[] = new long[H];
            long transv[] = new long[H];
            for (int i = 0; i < H; i++)
            {
                dpv[i] = dp[i][j];
                transv[i] = trans[i][j];
            }
            ntt(H, dpv, w);
            ntt(H, transv, w);
            for (int i = 0; i < H; i++)
            {
                dp[i][j] = dpv[i];
                trans[i][j] = transv[i];
            }
        }

        for (int i = 0; i < H; i++)
        {
            for (int j = 0; j < W; j++)
            {
                dp[i][j] = dp[i][j] * power(trans[i][j], T) % md;
            }
        }

        for (int j = 0; j < W; j++)
        {
            long dpv[] = new long[H];
            for (int i = 0; i < H; i++)
            {
                dpv[i] = dp[i][j];
            }
            intt(H, dpv, iw);
            for (int i = 0; i < H; i++)
            {
                dp[i][j] = dpv[i];
            }
        }
        intt(W, dp[c], iw);
        System.out.println(dp[c][d]);
    }
}
0