結果

問題 No.1753 Don't cheat.
ユーザー nok0nok0
提出日時 2021-06-20 00:37:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,083 ms / 3,000 ms
コード長 1,042 bytes
コンパイル時間 407 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 76,928 KB
最終ジャッジ日時 2024-06-10 06:52:10
合計ジャッジ時間 36,053 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,048 ms
76,544 KB
testcase_01 AC 1,045 ms
76,544 KB
testcase_02 AC 1,060 ms
76,416 KB
testcase_03 AC 1,058 ms
76,652 KB
testcase_04 AC 1,052 ms
76,656 KB
testcase_05 AC 1,053 ms
76,644 KB
testcase_06 AC 1,048 ms
76,676 KB
testcase_07 AC 1,050 ms
76,744 KB
testcase_08 AC 1,071 ms
76,544 KB
testcase_09 AC 1,050 ms
76,584 KB
testcase_10 AC 1,059 ms
76,672 KB
testcase_11 AC 1,047 ms
76,580 KB
testcase_12 AC 1,063 ms
76,612 KB
testcase_13 AC 1,083 ms
76,544 KB
testcase_14 AC 1,051 ms
76,544 KB
testcase_15 AC 1,047 ms
76,928 KB
testcase_16 AC 1,046 ms
76,544 KB
testcase_17 AC 1,070 ms
76,644 KB
testcase_18 AC 1,058 ms
76,552 KB
testcase_19 AC 1,048 ms
76,544 KB
testcase_20 AC 1,058 ms
76,544 KB
testcase_21 AC 1,054 ms
76,536 KB
testcase_22 AC 1,061 ms
76,872 KB
testcase_23 AC 1,059 ms
76,416 KB
testcase_24 AC 1,057 ms
76,924 KB
testcase_25 AC 1,044 ms
76,632 KB
testcase_26 AC 1,048 ms
76,596 KB
testcase_27 AC 1,050 ms
76,672 KB
testcase_28 AC 1,043 ms
76,672 KB
testcase_29 AC 1,044 ms
76,544 KB
testcase_30 AC 1,044 ms
76,544 KB
testcase_31 AC 1,043 ms
76,672 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353


def fwht(a) -> None:
    n = len(a)
    i = 1
    while i < n:
        for j in range(n):
            if (j & i) == 0:
                x, y = a[j], a[j | i]
                a[j] = (x + y) % MOD
                a[j | i] = (x - y) % MOD
        i *= 2


def ifwht(a) -> None:
    inv_2 = pow(2, MOD - 2, MOD)
    n = len(a)
    i = 1
    while i < n:
        for j in range(n):
            if (j & i) == 0:
                x, y = a[j], a[j | i]
                a[j] = (x + y) * inv_2 % MOD
                a[j | i] = (x - y) * inv_2 % MOD
        i *= 2


sz = 1 << 10
n = int(input())
a = list(map(int, input().split()))
while len(a) < sz:
    a.append(0)
s = sum(a)
sinv = pow(s, MOD - 2, MOD)
f = [0 for i in range(sz)]
f[0] = 1
for i in range(1, sz):
    f[i] = -a[i] * sinv % MOD
res = 0
for x in range(1, sz):
    g = f.copy()
    g[x] = 0
    fwht(g)
    for i in range(len(g)):
        g[i] = pow(g[i], MOD - 2, MOD)
    ifwht(g)
    res += g[x]
    res %= MOD
res *= a[0]
res %= MOD
res *= sinv
res %= MOD
print(res)
0