結果

問題 No.1540 級数おもちゃ
ユーザー dachengzdachengz
提出日時 2022-12-13 13:55:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 83 ms / 2,000 ms
コード長 709 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 81,904 KB
実行使用メモリ 86,272 KB
最終ジャッジ日時 2024-11-07 09:18:21
合計ジャッジ時間 4,620 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
51,840 KB
testcase_01 AC 39 ms
51,876 KB
testcase_02 AC 51 ms
62,592 KB
testcase_03 AC 60 ms
73,216 KB
testcase_04 AC 61 ms
72,832 KB
testcase_05 AC 67 ms
79,488 KB
testcase_06 AC 78 ms
84,608 KB
testcase_07 AC 78 ms
84,480 KB
testcase_08 AC 69 ms
80,256 KB
testcase_09 AC 59 ms
69,248 KB
testcase_10 AC 79 ms
84,608 KB
testcase_11 AC 80 ms
85,760 KB
testcase_12 AC 82 ms
85,632 KB
testcase_13 AC 82 ms
85,840 KB
testcase_14 AC 81 ms
85,904 KB
testcase_15 AC 80 ms
85,820 KB
testcase_16 AC 80 ms
85,632 KB
testcase_17 AC 79 ms
86,144 KB
testcase_18 AC 81 ms
85,888 KB
testcase_19 AC 81 ms
85,948 KB
testcase_20 AC 81 ms
86,180 KB
testcase_21 AC 58 ms
70,016 KB
testcase_22 AC 83 ms
85,760 KB
testcase_23 AC 65 ms
75,648 KB
testcase_24 AC 55 ms
65,536 KB
testcase_25 AC 54 ms
65,536 KB
testcase_26 AC 82 ms
86,016 KB
testcase_27 AC 53 ms
63,360 KB
testcase_28 AC 73 ms
82,192 KB
testcase_29 AC 54 ms
63,360 KB
testcase_30 AC 83 ms
86,272 KB
testcase_31 AC 40 ms
51,712 KB
testcase_32 AC 40 ms
51,840 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
N = int(input())
A = [int(_) for _ in input().split()]
inv = [1] * max(N + 1, 4)

for i in range(2, max(N + 1, 4)):
    inv[i] = mod - mod // i * inv[mod % i] % mod

ans = [0, 0, 0]
coef = [[4, mod - 2, 0], [0, 0, inv[3]]]
if N >= 1:
    ans[2] = (ans[2] + coef[1][2] * A[0]) % mod
if N >= 2:
    ans[0] = (ans[0] + coef[0][0] * A[1]) % mod
    ans[1] = (ans[1] + coef[0][1] * A[1]) % mod
for j, a in enumerate(A[2:], start=3):
    arr = coef[j & 1]
    mul = 4 * (j - 2) * inv[j - 1] % mod
    for i in range(3):
        arr[i] = arr[i] * mul % mod
    arr[1] = (arr[1] - 2 * inv[j - 1]) % mod
    for i in range(3):
        ans[i] = (ans[i] + arr[i] * a) % mod
print(ans[0], ans[1], ans[2])
0