結果

問題 No.663 セルオートマトンの逆操作
ユーザー konsin_tokage
提出日時 2018-03-11 10:59:32
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 97 ms / 2,000 ms
コード長 769 bytes
コンパイル時間 81 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-10-15 00:10:16
合計ジャッジ時間 2,034 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff #

from itertools import *
from collections import *
n = int(input())
e = [int(input()) for _ in range(n)]
zero = {(0,0,0),(1,0,0),(1,1,1)}
one = set(product(range(2), repeat=3)) - zero
table = [zero, one]
result = 0
for start_left, start_middle, start_right in table[e[0]]:
    state = Counter()
    state[start_middle, start_right] = 1
    for bit in e[1:-1]:
        temp = Counter()
        for left, middle, right in table[bit]:
            temp[middle, right] += state[left, middle]
            temp[middle, right] %= 10**9+7
        state = temp
    for (left, middle), count in state.items():
        if (middle == start_left
                and (left, middle, start_middle) in table[e[-1]]):
            result += count
            result %= 10**9+7
print(result)
0