結果

問題 No.663 セルオートマトンの逆操作
ユーザー fumofumofuni
提出日時 2023-06-21 01:40:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 83 ms / 2,000 ms
コード長 975 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,096 KB
実行使用メモリ 76,928 KB
最終ジャッジ日時 2024-06-28 12:22:52
合計ジャッジ時間 2,591 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff #

N=int(input())
A=[0]*N
for i in range(N):
    A[i]=int(input())
ans = 0
mod =1000000007
for i in range(2):
    for j in range(2):
        dp=[0]*4
        dp[i*2+j]=1
        for k in range(1,N-1):
            ndp=[0]*4
            if A[k]==0:
                ndp[3]+=dp[3]
                ndp[0]+=dp[0]
                ndp[0]+=dp[2]
            else :
                ndp[1]+=dp[0]
                ndp[2]+=dp[1]
                ndp[3]+=dp[1]
                ndp[1]+=dp[2]
                ndp[2]+=dp[3]
            for l in range(4):
                ndp[l]%=mod
                dp[l]=ndp[l]
        for k in range(4):
            p=1
            if i==0 and j==0:
                p=0
            if k%2==1 and i==1 and j==1:
                p=0
            q=1
            if k%2==0 and i==0:
                q=0
            if k//2==1 and k%2==1 and i==1:
                q=0
            if A[0]==p and A[N-1]==q:
                ans += dp[k]
             


print(ans%mod)
0