結果
問題 | No.147 試験監督(2) |
ユーザー | convexineq |
提出日時 | 2020-12-11 05:02:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 941 ms / 2,000 ms |
コード長 | 962 bytes |
コンパイル時間 | 367 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 78,908 KB |
最終ジャッジ日時 | 2024-09-19 21:06:32 |
合計ジャッジ時間 | 4,550 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 930 ms
78,720 KB |
testcase_01 | AC | 941 ms
78,720 KB |
testcase_02 | AC | 905 ms
78,908 KB |
testcase_03 | AC | 47 ms
62,208 KB |
ソースコード
def polymul(f,g): lf = len(f) lg = len(g) res = [0]*(lf+lg-1) for i in range(lf): for j in range(lg): res[i+j] += f[i]*g[j] res[i+j] %= MOD return res def fps_nth_term(f,g,N): assert g[0] != 0 while N: h = g[:] for i in range(1,len(g),2): h[i] = -h[i] f = polymul(f,h)[N%2:N+1:2] g = polymul(g,h)[:N+1:2] N //= 2 return f[0]*pow(g[0],MOD-2,MOD)%MOD # a[0],...,a[L-2] とL-1次特性多項式 g が与えられているL項間漸化式の第N項 def rec_nth_term(a,g,N): L = len(g) assert len(a) == L-1 f = polymul(a,g)[:L-1] return fps_nth_term(f,g,N) n = int(input()) MOD = 10**9+7 MOD2 = 10**9+6 ans = 1 for _ in range(n): c,d = input().split() r = 0 for i in d: r = (r*10+int(i))%MOD2 v = rec_nth_term([1,2],[1,-1,-1],int(c)) if v==0: ans = 0 ans *= pow(v,r,MOD) ans %= MOD print(ans%MOD)