結果

問題 No.1049 Zero (Exhaust)
ユーザー 👑 tamatotamato
提出日時 2020-05-08 21:48:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 100 ms / 2,000 ms
コード長 521 bytes
コンパイル時間 271 ms
コンパイル使用メモリ 87,628 KB
実行使用メモリ 91,812 KB
最終ジャッジ日時 2023-09-17 02:33:00
合計ジャッジ時間 3,638 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 70 ms
71,692 KB
testcase_01 AC 71 ms
71,672 KB
testcase_02 AC 70 ms
71,792 KB
testcase_03 AC 92 ms
86,788 KB
testcase_04 AC 71 ms
71,564 KB
testcase_05 AC 89 ms
84,740 KB
testcase_06 AC 99 ms
91,756 KB
testcase_07 AC 78 ms
77,940 KB
testcase_08 AC 84 ms
80,648 KB
testcase_09 AC 94 ms
87,828 KB
testcase_10 AC 94 ms
88,828 KB
testcase_11 AC 96 ms
88,816 KB
testcase_12 AC 97 ms
90,884 KB
testcase_13 AC 82 ms
79,564 KB
testcase_14 AC 87 ms
82,728 KB
testcase_15 AC 94 ms
86,900 KB
testcase_16 AC 91 ms
84,920 KB
testcase_17 AC 90 ms
85,432 KB
testcase_18 AC 96 ms
89,088 KB
testcase_19 AC 93 ms
87,124 KB
testcase_20 AC 82 ms
80,580 KB
testcase_21 AC 93 ms
87,444 KB
testcase_22 AC 86 ms
82,436 KB
testcase_23 AC 98 ms
91,812 KB
testcase_24 AC 100 ms
91,724 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    P, K = map(int, input().split())

    dp_0 = [0] * (K+1)
    dp_1 = [0] * (K+1)
    dp_0[0] = 1
    for i in range(K):
        # add
        dp_0[i+1] = (dp_0[i] + dp_1[i])%mod
        dp_1[i+1] = ((dp_0[i] + dp_1[i]) * (P-1))%mod
        # mult
        dp_0[i+1] = (dp_0[i+1] + (dp_0[i] * P)%mod + dp_1[i])%mod
        dp_1[i+1] = (dp_1[i+1] + dp_1[i] * (P-1))%mod
    print(dp_0[-1])


if __name__ == '__main__':
    main()
0