結果
問題 | No.314 ケンケンパ |
ユーザー |
![]() |
提出日時 | 2015-12-07 01:00:23 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 32 ms / 1,000 ms |
コード長 | 608 bytes |
コンパイル時間 | 209 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 10,752 KB |
最終ジャッジ日時 | 2024-09-14 17:11:17 |
合計ジャッジ時間 | 1,486 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
n = int(input()) mat = [[0,0,1],[1,0,0],[1,1,0]] mod = 1000000007 ret = [[1,0,0],[0,1,0],[0,0,1]] while n>0 : if (n&1) == 1 : tmp = [[0,0,0],[0,0,0],[0,0,0]] for i in range(3) : for k in range(3) : for j in range(3) : tmp[i][j] += ret[i][k] * mat[k][j] if tmp[i][j] >= mod : tmp[i][j] %= mod ret = tmp tmp_ = [[0,0,0],[0,0,0],[0,0,0]] for i in range(3) : for k in range(3) : for j in range(3) : tmp_[i][j] += mat[i][k] * mat[k][j] if tmp_[i][j] >= mod : tmp_[i][j] %= mod mat = tmp_ n = n>>1 ans = (ret[0][2] + ret[1][2] + ret[2][2]) % mod print(ans)