結果

問題 No.1049 Zero (Exhaust)
コンテスト
ユーザー convexineq
提出日時 2020-05-08 21:55:19
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,112 ms / 2,000 ms
コード長 333 bytes
コンパイル時間 107 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-07-04 00:36:03
合計ジャッジ時間 16,267 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# coding: utf-8
# Your code here!
import sys
read = sys.stdin.read
readline = sys.stdin.readline 
readlines = sys.stdin.readlines

p,k = [int(i) for i in read().split()]


a = 1 # zero
b = 0 # non-zero
MOD = 10**9+7

dp = [1,0,0,0,0]
for i in range(k):
    a,b = ((a+b*(p-1))+a*p+b*(p-1))%MOD, ((a+b*(p-1))+b*(p-1))%MOD
    
print(a)
0