結果

問題 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

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()
0