#995475 (Python3) No.1049 Zero (Exhaust)

• 問題: No.1049 Zero (Exhaust)
• ユーザー: vwxyz
• 提出日時: 2024-07-17 03:53:02
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 1,025 ms / 2,000 ms
• コード長: 158 bytes
• コンパイル時間: 110 ms
• コンパイル使用メモリ: 12,416 KB
• 実行使用メモリ: 10,624 KB
• 最終ジャッジ日時: 2024-07-17 03:53:17
• 合計ジャッジ時間: 14,775 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	30 ms; 10,496 KB
testcase_01	AC	30 ms; 10,496 KB
testcase_02	AC	29 ms; 10,496 KB
testcase_03	AC	660 ms; 10,624 KB
testcase_04	AC	30 ms; 10,496 KB
testcase_05	AC	571 ms; 10,496 KB
testcase_06	AC	939 ms; 10,496 KB
testcase_07	AC	141 ms; 10,496 KB
testcase_08	AC	293 ms; 10,496 KB
testcase_09	AC	695 ms; 10,496 KB
testcase_10	AC	761 ms; 10,624 KB
testcase_11	AC	760 ms; 10,496 KB
testcase_12	AC	869 ms; 10,496 KB
testcase_13	AC	220 ms; 10,624 KB
testcase_14	AC	435 ms; 10,496 KB
testcase_15	AC	650 ms; 10,624 KB
testcase_16	AC	537 ms; 10,496 KB
testcase_17	AC	558 ms; 10,496 KB
testcase_18	AC	807 ms; 10,496 KB
testcase_19	AC	686 ms; 10,496 KB
testcase_20	AC	308 ms; 10,496 KB
testcase_21	AC	705 ms; 10,496 KB
testcase_22	AC	425 ms; 10,496 KB
testcase_23	AC	1,025 ms; 10,624 KB
testcase_24	AC	950 ms; 10,496 KB

ソースコード

P,K=map(int,input().split())
mod=10**9+7
dp0,dp1=1,0
for k in range(K):
    dp0,dp1=(dp0+dp1+dp0*P+dp1)%mod,((dp0+dp1)*(P-1)+dp1*(P-1))%mod
ans=dp0
print(ans)