結果
問題 | No.1704 Many Bus Stops (easy) |
ユーザー |
|
提出日時 | 2021-10-08 22:56:15 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 616 ms / 2,000 ms |
コード長 | 1,183 bytes |
コンパイル時間 | 167 ms |
コンパイル使用メモリ | 82,408 KB |
実行使用メモリ | 78,848 KB |
最終ジャッジ日時 | 2024-07-23 06:09:33 |
合計ジャッジ時間 | 18,516 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 41 |
ソースコード
import sysinput = lambda :sys.stdin.readline()[:-1]ni = lambda :int(input())na = lambda :list(map(int,input().split()))sys.setrecursionlimit(10**7)yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")########################################################################a*bの計算def mat_mul(a,b):n,m,l=len(a),len(b[0]),len(b)c=[[0]*m for _ in range(n)]for i in range(n):for k in range(l):for j in range(m):c[i][j]=(c[i][j]+a[i][k]*b[k][j])%modreturn c#a^nの計算def mat_pow(a,n):m=len(a)b=[[0]*m for _ in range(m)]for i in range(m):b[i][i]=1while n>0:if n&1:b=mat_mul(b,a)a=mat_mul(a,a)n>>=1return bt = ni()p = mod = 10**9+7z = pow(3,p-2,p)s = list(zip([z,0,0,1,0,0]))b = [[z,0,0,0,z,z],[0,z,0,z,0,z],[0,0,z,z,z,0],[1,0,0,0,0,0],[0,1,0,0,0,0],[0,0,1,0,0,0]]for n in [ni() for i in range(t)]:if n==1:print(z)else:print(mat_mul(mat_pow(b,n-1),s)[0][0]%mod)