結果
問題 | No.1706 Many Bus Stops (hard) |
ユーザー |
|
提出日時 | 2021-10-08 23:44:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 52 ms / 2,000 ms |
コード長 | 1,183 bytes |
コンパイル時間 | 698 ms |
コンパイル使用メモリ | 82,340 KB |
実行使用メモリ | 64,956 KB |
最終ジャッジ日時 | 2024-07-23 08:10:07 |
合計ジャッジ時間 | 3,671 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
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 bc,n,m = na()p = mod = 10**9+7z = pow(c,p-2,p)s = list(zip([z,0,1,0]))b = [[z,0,0,z],[0,z,z*(c-1),z*(c-2)],[1,0,0,0],[0,1,0,0]]if n==1:print((1-pow(1-z, m, mod))%mod)else:p = mat_mul(mat_pow(b,n-1),s)#print(mat_mul(mat_pow(b,n-1),s))print((1-pow(1-mat_mul(mat_pow(b,n-1),s)[0][0],m,mod))%mod)