結果
問題 |
No.314 ケンケンパ
|
ユーザー |
![]() |
提出日時 | 2015-12-07 00:56:46 |
言語 | PyPy2 (7.3.15) |
結果 |
AC
|
実行時間 | 76 ms / 1,000 ms |
コード長 | 603 bytes |
コンパイル時間 | 1,070 ms |
コンパイル使用メモリ | 76,444 KB |
実行使用メモリ | 75,688 KB |
最終ジャッジ日時 | 2024-09-14 17:10:58 |
合計ジャッジ時間 | 3,648 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
n = 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)