結果
問題 | No.492 IOI数列 |
ユーザー |
|
提出日時 | 2017-03-10 23:35:20 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 30 ms / 1,000 ms |
コード長 | 899 bytes |
コンパイル時間 | 163 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 11,008 KB |
最終ジャッジ日時 | 2024-06-24 08:54:52 |
合計ジャッジ時間 | 1,626 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
def mul_mat_mod(A, B, mod): n = len(A) m = len(A[0]) p = len(B[0]) res = [[0 for j in range(p)] for i in range(n)] for i in range(n): for j in range(p): for k in range(m): res[i][j] = res[i][j]+A[i][k]*B[k][j] res[i][j] %= mod return res def pow_mat_mod(A, k, mod): n = len(A) res = [[0 for j in range(n)] for i in range(n)] B = [[A[i][j] for j in range(n)] for i in range(n)] for i in range(n): res[i][i] = 1 while k > 0: if k % 2 != 0: res = mul_mat_mod(res, B, mod) k = k//2 B = mul_mat_mod(B, B, mod) return res N = int(input()) mod_a = 1000000007 mod_b = 101010101010101010101 P = [[100,1],[0,1]] A = pow_mat_mod(P, N-1, mod_a) B = pow_mat_mod(P, N-1, mod_b) x = A[0][0]*1+A[0][1]*1 y = B[0][0]*1+B[0][1]*1 x %= mod_a y %= mod_b print(x) print(y)