結果

問題 No.1753 Don't cheat.
ユーザー nok0nok0
提出日時 2021-06-20 00:37:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,173 ms / 3,000 ms
コード長 1,042 bytes
コンパイル時間 277 ms
コンパイル使用メモリ 87,300 KB
実行使用メモリ 78,304 KB
最終ジャッジ日時 2023-08-30 06:22:30
合計ジャッジ時間 39,865 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,162 ms
77,708 KB
testcase_01 AC 1,153 ms
78,028 KB
testcase_02 AC 1,156 ms
77,648 KB
testcase_03 AC 1,159 ms
77,680 KB
testcase_04 AC 1,153 ms
77,984 KB
testcase_05 AC 1,152 ms
77,460 KB
testcase_06 AC 1,145 ms
77,880 KB
testcase_07 AC 1,148 ms
77,956 KB
testcase_08 AC 1,150 ms
77,840 KB
testcase_09 AC 1,149 ms
78,036 KB
testcase_10 AC 1,150 ms
78,000 KB
testcase_11 AC 1,153 ms
77,980 KB
testcase_12 AC 1,151 ms
78,100 KB
testcase_13 AC 1,152 ms
78,060 KB
testcase_14 AC 1,154 ms
78,092 KB
testcase_15 AC 1,149 ms
77,960 KB
testcase_16 AC 1,149 ms
78,088 KB
testcase_17 AC 1,173 ms
77,964 KB
testcase_18 AC 1,167 ms
78,000 KB
testcase_19 AC 1,153 ms
78,228 KB
testcase_20 AC 1,161 ms
77,684 KB
testcase_21 AC 1,165 ms
77,896 KB
testcase_22 AC 1,171 ms
78,136 KB
testcase_23 AC 1,157 ms
77,796 KB
testcase_24 AC 1,155 ms
78,208 KB
testcase_25 AC 1,154 ms
78,084 KB
testcase_26 AC 1,150 ms
77,916 KB
testcase_27 AC 1,151 ms
77,792 KB
testcase_28 AC 1,152 ms
78,284 KB
testcase_29 AC 1,150 ms
78,304 KB
testcase_30 AC 1,143 ms
77,796 KB
testcase_31 AC 1,143 ms
77,920 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