結果

問題 No.314 ケンケンパ
ユーザー tanimani364
提出日時 2021-03-21 19:25:29
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 627 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 65,792 KB
最終ジャッジ日時 2024-11-22 14:30:29
合計ジャッジ時間 2,488 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 3
other RE * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
#coding:utf-8
from __future__ import unicode_literals
import math
from sys import stdin
import numpy as np
# from matplotlib import pyplot as plt



def main():
	read=stdin.readline

	#edit here!
	n=int(read())
	m=int(1000000000+7)

	dp=np.zeros((n+1,3))
	dp[0][0]=1
	for i in range(n):
		for j in range(3):
			if j==0:
				dp[i+1][1]+=dp[i][0]
				dp[i+1][1]%=m
			elif j==1:
				dp[i+1][2]+=dp[i][1]
				dp[i+1][2]%=m
				dp[i+1][0]+=dp[i][1]
				dp[i+1][0]%=m
			else:
				dp[i+1][0]+=dp[i][j]
				dp[i+1][0]%=m
	
	ans=(dp[n][0]+dp[n][1]+dp[n][2])%m
	print(ans)

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