結果

問題 No.1241 Eternal Tours
ユーザー hitonanodehitonanode
提出日時 2020-09-10 01:54:20
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 3,029 ms / 6,000 ms
コード長 2,129 bytes
コンパイル時間 342 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 107,228 KB
最終ジャッジ日時 2024-05-09 23:15:13
合計ジャッジ時間 67,561 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 521 ms
46,272 KB
testcase_01 AC 525 ms
46,144 KB
testcase_02 AC 2,150 ms
106,856 KB
testcase_03 AC 526 ms
46,016 KB
testcase_04 AC 524 ms
46,020 KB
testcase_05 AC 529 ms
46,276 KB
testcase_06 AC 533 ms
45,880 KB
testcase_07 AC 521 ms
46,016 KB
testcase_08 AC 525 ms
46,668 KB
testcase_09 AC 525 ms
46,272 KB
testcase_10 AC 527 ms
46,276 KB
testcase_11 AC 522 ms
46,020 KB
testcase_12 AC 522 ms
46,148 KB
testcase_13 AC 527 ms
46,404 KB
testcase_14 AC 1,613 ms
71,992 KB
testcase_15 AC 524 ms
46,028 KB
testcase_16 AC 765 ms
52,032 KB
testcase_17 AC 2,902 ms
106,948 KB
testcase_18 AC 2,654 ms
102,860 KB
testcase_19 AC 2,797 ms
98,404 KB
testcase_20 AC 596 ms
46,020 KB
testcase_21 AC 556 ms
46,400 KB
testcase_22 AC 2,693 ms
98,956 KB
testcase_23 AC 582 ms
47,368 KB
testcase_24 AC 535 ms
46,532 KB
testcase_25 AC 528 ms
46,532 KB
testcase_26 AC 531 ms
45,896 KB
testcase_27 AC 527 ms
46,668 KB
testcase_28 AC 2,646 ms
106,732 KB
testcase_29 AC 2,134 ms
107,128 KB
testcase_30 AC 2,110 ms
106,720 KB
testcase_31 AC 1,383 ms
76,096 KB
testcase_32 AC 2,732 ms
107,056 KB
testcase_33 AC 2,861 ms
98,900 KB
testcase_34 AC 2,760 ms
107,228 KB
testcase_35 AC 2,866 ms
106,944 KB
testcase_36 AC 530 ms
46,268 KB
testcase_37 AC 531 ms
46,024 KB
testcase_38 AC 2,960 ms
99,164 KB
testcase_39 AC 2,939 ms
98,620 KB
testcase_40 AC 2,949 ms
107,096 KB
testcase_41 AC 2,990 ms
106,920 KB
testcase_42 AC 2,944 ms
98,744 KB
testcase_43 AC 3,029 ms
106,716 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
import numpy as np


class NTT():
    def __init__(self, D, MOD, root):
        self.md = MOD
        self.w = np.array([1], np.int64)
        self.iw = np.array([1], np.int64)

        while len(self.w) < 1 << (D - 1):
            dw = pow(root, (self.md - 1) // (len(self.w) * 4), self.md)
            dwinv = pow(dw, -1, self.md)
            self.w = np.r_[self.w, self.w * dw] % self.md
            self.iw = np.r_[self.iw, self.iw * dwinv] % self.md

    def ntt(self, mat):
        in_shape = mat.shape
        n = in_shape[-1]

        m = n // 2
        while m:
            mat = mat.reshape(-1, n // (m * 2), 2, m)
            w_use = self.w[:n // (m * 2)].reshape(1, -1, 1)
            y = mat[:, :, 1] * w_use % self.md
            mat = np.stack((mat[:, :, 0] + y, mat[:, :, 0] + self.md - y), 2) % self.md
            m //= 2
        return mat.reshape(in_shape)

    def intt(self, mat):
        in_shape = mat.shape
        n = in_shape[-1]

        m = 1
        while m < n:
            mat = mat.reshape(-1, n // (m * 2), 2, m)
            iw_use = self.iw[:n // (m * 2)].reshape(1, -1, 1)
            mat = np.stack((mat[:, :, 0] + mat[:, :, 1], (mat[:, :, 0] + self.md - mat[:, :, 1]) * iw_use), 2) % self.md
            m *= 2
        n_inv = pow(n, -1, self.md)
        return mat.reshape(in_shape) * n_inv % self.md


X, Y, T, a, b, c, d = list(map(int, input().split()))

md = 998244353
T = (T - 1) % (md - 1) + 1

H, W = 1 << (X + 1), 1 << (Y + 1)
dp = np.zeros((H, W), np.int64)
trans = np.zeros((H, W), np.int64)

dp[a, b] = dp[-a, -b] = 1
dp[a, -b] = dp[-a, b] = md - 1
trans[0, 0] = trans[0, 1] = trans[0, -1] = trans[1, 0] = trans[-1, 0] = 1

ntt = NTT(18, md, 3)

dp = ntt.ntt(dp)
trans = ntt.ntt(trans)

dp = dp.T
trans = trans.T

dp = ntt.ntt(dp)
trans = ntt.ntt(trans)

def matpow(x, n, mod):
    ret, tmp = np.ones(x.shape, np.int64), x % mod
    while n:
        if n % 2:
            ret = ret * tmp % mod
        tmp = tmp * tmp % mod
        n //= 2
    return ret

dp = dp * matpow(trans, T, md) % md

dp = ntt.intt(dp)
dp = dp.T
dp = ntt.intt(dp)

print(dp[c][d])
0