結果
問題 | No.2883 K-powered Sum of Fibonacci |
ユーザー | miya145592 |
提出日時 | 2024-09-08 22:08:31 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,377 ms / 3,000 ms |
コード長 | 1,510 bytes |
コンパイル時間 | 316 ms |
コンパイル使用メモリ | 82,368 KB |
実行使用メモリ | 77,424 KB |
最終ジャッジ日時 | 2024-09-08 22:08:51 |
合計ジャッジ時間 | 19,581 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 42 ms
53,552 KB |
testcase_01 | AC | 40 ms
53,224 KB |
testcase_02 | AC | 45 ms
61,860 KB |
testcase_03 | AC | 45 ms
60,992 KB |
testcase_04 | AC | 44 ms
59,560 KB |
testcase_05 | AC | 142 ms
76,260 KB |
testcase_06 | AC | 50 ms
63,280 KB |
testcase_07 | AC | 405 ms
76,280 KB |
testcase_08 | AC | 127 ms
76,168 KB |
testcase_09 | AC | 84 ms
71,688 KB |
testcase_10 | AC | 82 ms
72,868 KB |
testcase_11 | AC | 781 ms
76,676 KB |
testcase_12 | AC | 126 ms
76,028 KB |
testcase_13 | AC | 79 ms
72,932 KB |
testcase_14 | AC | 930 ms
76,520 KB |
testcase_15 | AC | 273 ms
76,236 KB |
testcase_16 | AC | 46 ms
62,204 KB |
testcase_17 | AC | 159 ms
76,196 KB |
testcase_18 | AC | 68 ms
70,260 KB |
testcase_19 | AC | 181 ms
76,108 KB |
testcase_20 | AC | 1,176 ms
76,924 KB |
testcase_21 | AC | 1,075 ms
77,180 KB |
testcase_22 | AC | 1,215 ms
77,052 KB |
testcase_23 | AC | 921 ms
77,104 KB |
testcase_24 | AC | 920 ms
76,640 KB |
testcase_25 | AC | 1,290 ms
77,248 KB |
testcase_26 | AC | 1,345 ms
77,164 KB |
testcase_27 | AC | 1,304 ms
77,176 KB |
testcase_28 | AC | 1,314 ms
77,160 KB |
testcase_29 | AC | 1,371 ms
77,404 KB |
testcase_30 | AC | 59 ms
66,224 KB |
testcase_31 | AC | 47 ms
62,168 KB |
testcase_32 | AC | 49 ms
62,276 KB |
testcase_33 | AC | 93 ms
70,836 KB |
testcase_34 | AC | 47 ms
60,124 KB |
testcase_35 | AC | 46 ms
62,416 KB |
testcase_36 | AC | 50 ms
63,704 KB |
testcase_37 | AC | 51 ms
62,624 KB |
testcase_38 | AC | 50 ms
62,524 KB |
testcase_39 | AC | 50 ms
62,652 KB |
testcase_40 | AC | 40 ms
53,508 KB |
testcase_41 | AC | 39 ms
53,272 KB |
testcase_42 | AC | 1,377 ms
77,424 KB |
ソースコード
# https://tech.aru-zakki.com/python-pow-matrix/ def matmul(A, B): N = len(A) K = len(A[0]) M = len(B[0]) c = [[0 for _ in range(M)] for _ in range(N)] for i in range(N) : for j in range(K) : for k in range(M) : c[i][k] += A[i][j] * B[j][k] % MOD c[i][k] %= MOD return c def pow_matrix(A, p): n = len(A) # 単位行列 c = [[1 if i == j else 0 for i in range(n)] for j in range(n)] while p > 0 : if p%2 == 1 : c = matmul(c, A) A = matmul(A, A) p //= 2 return c def nPr(n, r, mod): if ( r<0 or r>n ): return 0 return g1[n] * g2[n-r] % mod def nCr(n, r, mod): if ( r<0 or r>n ): return 0 r = min(r, n-r) return (g1[n] * g2[r] % mod) * g2[n-r] % mod import sys input = sys.stdin.readline MOD = 998244353 N, K = map(int, input().split()) g1 = [1, 1] # 元テーブル g2 = [1, 1] #逆元テーブル inverse = [0, 1] #逆元テーブル計算用テーブル fact = [1, 1] for i in range( 2, K + 1 ): g1.append( ( g1[-1] * i ) % MOD ) inverse.append( ( -inverse[MOD % i] * (MOD//i) ) % MOD ) g2.append( (g2[-1] * inverse[-1]) % MOD ) fact.append( (fact[-1] * i) % MOD ) A = [[1 for _ in range(K+2)]] C = [[0 for _ in range(K+2)] for _ in range(K+2)] for i in range(K+1): for j in range(K+1): C[i][j] = nCr(K-j, K-i-j, MOD) C[0][-1] = 1 C[-1][-1] = 1 CC = pow_matrix(C, N-1) ans = matmul(A, CC) print(ans[0][-1])