結果
問題 | No.980 Fibonacci Convolution Hard |
ユーザー | onakasuitacity |
提出日時 | 2020-11-06 23:10:22 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 750 ms / 2,000 ms |
コード長 | 1,391 bytes |
コンパイル時間 | 349 ms |
コンパイル使用メモリ | 82,200 KB |
実行使用メモリ | 92,456 KB |
最終ジャッジ日時 | 2024-07-22 13:34:46 |
合計ジャッジ時間 | 16,817 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 725 ms
92,032 KB |
testcase_01 | AC | 716 ms
92,032 KB |
testcase_02 | AC | 739 ms
91,968 KB |
testcase_03 | AC | 750 ms
91,904 KB |
testcase_04 | AC | 744 ms
92,116 KB |
testcase_05 | AC | 711 ms
92,032 KB |
testcase_06 | AC | 723 ms
91,916 KB |
testcase_07 | AC | 750 ms
92,456 KB |
testcase_08 | AC | 705 ms
92,032 KB |
testcase_09 | AC | 696 ms
91,904 KB |
testcase_10 | AC | 747 ms
92,044 KB |
testcase_11 | AC | 721 ms
91,876 KB |
testcase_12 | AC | 728 ms
92,032 KB |
testcase_13 | AC | 738 ms
92,288 KB |
testcase_14 | AC | 730 ms
91,776 KB |
testcase_15 | AC | 709 ms
92,080 KB |
testcase_16 | AC | 695 ms
91,904 KB |
ソースコード
import sys INF = 1 << 60 MOD = 10**9 + 7 # 998244353 sys.setrecursionlimit(2147483647) input = lambda:sys.stdin.readline().rstrip() def linear_recurrence(A): n = len(A) C, pC = [1], [0] m, pd = 0, 0 for i in range(n): m += 1 d = sum(c * A[i - j] % MOD for j, c in enumerate(C)) % MOD if d == 0: continue q = pow(pd, MOD - 2, MOD) * d % MOD if len(C) < len(pC) + m: T = C[:] C += [0] * (len(pC) + m - len(C)) for j, v in enumerate(pC): C[j + m] = (C[j + m] - q * v) % MOD pC = T m, pd = 0, d else: for j, v in enumerate(pC): C[j + m] = (C[j + m] - q * v) % MOD return [-v % MOD for v in C[1:]] def resolve(): p = int(input()) dp = [0] * 10 dp[1] = 1 for i in range(2, len(dp)): dp[i] = (p * dp[i - 1] + dp[i - 2]) % MOD dp2 = [0] * (len(dp) * 2 - 1) for i in range(len(dp)): for j in range(len(dp)): dp2[i + j] += dp[i] * dp[j] dp2[i + j] %= MOD C = linear_recurrence(dp2[:len(dp)]) l = len(C) M = 2 * 10**6 dp = [0] * M dp[:l] = dp2[:l] for i in range(l, M): dp[i] = sum(c * dp[i - j] % MOD for j, c in enumerate(C, 1)) % MOD for _ in range(int(input())): print(dp[int(input()) - 2]) resolve()