結果
問題 |
No.663 セルオートマトンの逆操作
|
ユーザー |
![]() |
提出日時 | 2020-09-03 02:28:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 76 ms / 2,000 ms |
コード長 | 1,334 bytes |
コンパイル時間 | 220 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 70,400 KB |
最終ジャッジ日時 | 2024-11-22 08:04:40 |
合計ジャッジ時間 | 3,110 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 29 |
ソースコード
import sys def input(): return sys.stdin.readline().strip() def list2d(a, b, c): return [[c] * b for i in range(a)] def list3d(a, b, c, d): return [[[d] * c for j in range(b)] for i in range(a)] def list4d(a, b, c, d, e): return [[[[e] * d for j in range(c)] for j in range(b)] for i in range(a)] def ceil(x, y=1): return int(-(-x // y)) def INT(): return int(input()) def MAP(): return map(int, input().split()) def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)] def Yes(): print('Yes') def No(): print('No') def YES(): print('YES') def NO(): print('NO') sys.setrecursionlimit(10**9) INF = 10**19 MOD = 10**9 + 7 EPS = 10**-10 N = INT() B = LIST(N) # 000,100,111は0 is_0 = set([0, 4, 7]) dp = list3d(1<<2, 1<<3, N, 0) for bit in range(1<<3): if (bit in is_0) ^ B[0]: dp[bit>>1][bit][0] = 1 msk = (1<<3) - 1 for i in range(1, N): for init in range(1<<2): for bit in range(1<<3): for k in range(2): nxt = ((bit<<1)&msk) | k if (nxt in is_0) ^ B[i]: dp[init][nxt][i] += dp[init][bit][i-1] dp[init][nxt][i] %= MOD ans = 0 msk = (1<<2) - 1 for init in range(1<<2): for bit in range(1<<3): if init == (bit&msk): ans += dp[init][bit][-1] ans %= MOD print(ans)