結果
問題 | No.1050 Zero (Maximum) |
ユーザー | convexineq |
提出日時 | 2020-05-08 22:03:00 |
言語 | Python3 (3.11.6 + numpy 1.26.0 + scipy 1.11.3) |
結果 |
AC
|
実行時間 | 1,015 ms / 2,000 ms |
コード長 | 966 bytes |
コンパイル時間 | 78 ms |
コンパイル使用メモリ | 10,920 KB |
実行使用メモリ | 8,608 KB |
最終ジャッジ日時 | 2023-09-17 02:50:28 |
合計ジャッジ時間 | 6,849 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 16 ms
8,208 KB |
testcase_01 | AC | 18 ms
8,296 KB |
testcase_02 | AC | 197 ms
8,436 KB |
testcase_03 | AC | 144 ms
8,280 KB |
testcase_04 | AC | 598 ms
8,392 KB |
testcase_05 | AC | 661 ms
8,544 KB |
testcase_06 | AC | 281 ms
8,264 KB |
testcase_07 | AC | 348 ms
8,352 KB |
testcase_08 | AC | 35 ms
8,312 KB |
testcase_09 | AC | 173 ms
8,384 KB |
testcase_10 | AC | 845 ms
8,388 KB |
testcase_11 | AC | 590 ms
8,540 KB |
testcase_12 | AC | 23 ms
8,356 KB |
testcase_13 | AC | 16 ms
8,312 KB |
testcase_14 | AC | 16 ms
8,212 KB |
testcase_15 | AC | 16 ms
8,308 KB |
testcase_16 | AC | 924 ms
8,556 KB |
testcase_17 | AC | 1,015 ms
8,608 KB |
ソースコード
MOD=10**9+7 def matmul(A,B,mod): # A,B: 行列 res = [[0]*len(B[0]) for _ in [None]*len(A)] for i, resi in enumerate(res): for k, aik in enumerate(A[i]): for j,bkj in enumerate(B[k]): resi[j] += aik*bkj resi[j] %= mod return res def matpow(A,p,M): #A^p mod M if p%2: return matmul(A, matpow(A,p-1,M), M) elif p > 0: b = matpow(A,p//2,M) return matmul(b,b,M) else: return [[1 if i == j else 0 for j in range(len(A))] for i in range(len(A))] # coding: utf-8 # Your code here! import sys read = sys.stdin.read readline = sys.stdin.readline readlines = sys.stdin.readlines m,k = [int(i) for i in read().split()] mat = [[0]*m for _ in range(m)] for i in range(m): for j in range(m): mat[(i+j)%m][i] += 1 mat[(i*j)%m][i] += 1 #print(mat) v = [[0] for _ in range(m)] v[0][0] = 1 v = matmul(matpow(mat,k,MOD), v,MOD) print(v[0][0])