結果

問題 No.980 Fibonacci Convolution Hard
ユーザー onakasuitacityonakasuitacity
提出日時 2020-11-06 23:10:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 618 ms / 2,000 ms
コード長 1,391 bytes
コンパイル時間 265 ms
コンパイル使用メモリ 86,968 KB
実行使用メモリ 93,664 KB
最終ジャッジ日時 2023-09-29 19:30:47
合計ジャッジ時間 15,186 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 601 ms
93,500 KB
testcase_01 AC 605 ms
93,580 KB
testcase_02 AC 557 ms
93,388 KB
testcase_03 AC 594 ms
93,104 KB
testcase_04 AC 565 ms
93,476 KB
testcase_05 AC 563 ms
93,580 KB
testcase_06 AC 579 ms
93,572 KB
testcase_07 AC 600 ms
93,644 KB
testcase_08 AC 588 ms
93,252 KB
testcase_09 AC 590 ms
93,664 KB
testcase_10 AC 596 ms
93,564 KB
testcase_11 AC 591 ms
93,492 KB
testcase_12 AC 590 ms
93,088 KB
testcase_13 AC 600 ms
93,396 KB
testcase_14 AC 618 ms
93,448 KB
testcase_15 AC 587 ms
93,496 KB
testcase_16 AC 556 ms
93,396 KB
権限があれば一括ダウンロードができます

ソースコード

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