結果
問題 | No.840 ほむほむほむら |
ユーザー | ryuhei1230 |
提出日時 | 2019-08-20 00:19:29 |
言語 | PyPy3 (7.3.13) |
結果 |
AC
|
実行時間 | 2,208 ms / 4,000 ms |
コード長 | 1,053 bytes |
コンパイル時間 | 286 ms |
コンパイル使用メモリ | 87,268 KB |
実行使用メモリ | 78,324 KB |
最終ジャッジ日時 | 2023-07-28 05:11:38 |
合計ジャッジ時間 | 15,027 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 82 ms
76,060 KB |
testcase_01 | AC | 84 ms
76,472 KB |
testcase_02 | AC | 120 ms
76,504 KB |
testcase_03 | AC | 391 ms
76,392 KB |
testcase_04 | AC | 81 ms
76,140 KB |
testcase_05 | AC | 82 ms
76,144 KB |
testcase_06 | AC | 88 ms
76,532 KB |
testcase_07 | AC | 174 ms
76,556 KB |
testcase_08 | AC | 635 ms
77,620 KB |
testcase_09 | AC | 91 ms
76,508 KB |
testcase_10 | AC | 82 ms
76,448 KB |
testcase_11 | AC | 98 ms
76,508 KB |
testcase_12 | AC | 227 ms
76,664 KB |
testcase_13 | AC | 1,464 ms
77,944 KB |
testcase_14 | AC | 256 ms
77,328 KB |
testcase_15 | AC | 83 ms
76,504 KB |
testcase_16 | AC | 103 ms
76,564 KB |
testcase_17 | AC | 398 ms
77,320 KB |
testcase_18 | AC | 1,827 ms
77,716 KB |
testcase_19 | AC | 2,208 ms
78,000 KB |
testcase_20 | AC | 73 ms
71,412 KB |
testcase_21 | AC | 81 ms
76,472 KB |
testcase_22 | AC | 106 ms
76,084 KB |
testcase_23 | AC | 2,113 ms
78,324 KB |
testcase_24 | AC | 109 ms
76,824 KB |
testcase_25 | AC | 78 ms
76,560 KB |
testcase_26 | AC | 124 ms
76,500 KB |
testcase_27 | AC | 2,116 ms
78,264 KB |
ソースコード
def mul(A, B) : N=len(A) mod=998244353 C=[[0 for i in range(N)] for j in range(N)] for i in range (N) : for j in range(N) : for k in range(N) : C[i][j]+=A[i][k]*B[k][j]%mod C[i][j]%=mod return C def powmat(A, n) : N=len(A) C=[[0 for i in range(N)] for j in range(N)] for i in range(N) : C[i][i] = 1 while n > 0 : if n%2==1 : C = mul(C,A) A=mul(A,A) n>>=1 return C def solve() : N, K = map(int,input().split()) mod = 998244353 M = K**3 A = [[0 for i in range(M)] for j in range(M)] for i in range(K) : for j in range(K) : for k in range(K) : cur=i*K*K+j*K+k; nxt1=i*K*K+j*K+(k+1)%K; nxt2=i*K*K+(j+k)%K*K+k; nxt3=(i+j)%K*K*K+j*K+k; A[nxt1][cur]+=1 A[nxt2][cur]+=1 A[nxt3][cur]+=1 C = powmat(A, N) ans = 0 for i in range(K*K) : ans += C[0][i] print(ans%mod) solve()