結果

問題 No.314 ケンケンパ
ユーザー tanimani364
提出日時 2021-03-21 19:17:46
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 627 bytes
コンパイル時間 118 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 90,936 KB
最終ジャッジ日時 2024-11-22 13:53:09
合計ジャッジ時間 18,508 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 TLE * 4
権限があれば一括ダウンロードができます

ソースコード

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=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(int(ans))

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