結果
| 問題 |
No.2283 Prohibit Three Consecutive
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2023-04-28 23:35:19 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 1,739 ms / 2,000 ms |
| コード長 | 2,266 bytes |
| コンパイル時間 | 222 ms |
| コンパイル使用メモリ | 82,580 KB |
| 実行使用メモリ | 278,824 KB |
| 最終ジャッジ日時 | 2024-11-17 22:27:34 |
| 合計ジャッジ時間 | 8,830 ms |
|
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | AC * 13 |
ソースコード
import sys
from collections import deque, Counter
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
inf = 2 ** 63 - 1
mod = 998244353
def solve():
n = ii()
s = [-1 if v == '?' else int(v) for v in input()]
p = []
for i in range(n):
if s[i] == -1:
p.append(i)
p = list(set(p[:2] + p[-2:]))
m = len(p)
for bit in range(1 << m):
a = s[::]
for i in range(m):
if 1 & (bit >> i):
a[p[i]] = 1
else:
a[p[i]] = 0
dp = [[[0] * 2 for _ in range(2)] for _ in range(n + 1)]
if a[0] == a[1] == -1:
if a[2] != 0:
dp[2][0][0] = 1
dp[2][0][1] = 1
dp[2][1][0] = 1
if a[2] != 1:
dp[2][1][1] = 1
elif a[0] == -1:
if not a[1] == a[2] == 0:
dp[2][0][a[1]] = 1
if not a[1] == a[2] == 1:
dp[2][1][a[1]] = 1
elif a[1] == -1:
if not a[0] == a[2] == 0:
dp[2][a[0]][0] = 1
if not a[0] == a[2] == 1:
dp[2][a[0]][1] = 1
else:
if not a[0] == a[1] == a[2]:
dp[2][a[0]][a[1]] = 1
for i in range(2, n):
for j in range(2):
for k in range(2):
if a[i] == -1:
if j == k:
dp[i + 1][k][1 - k] |= dp[i][j][k]
else:
dp[i + 1][k][0] |= dp[i][j][k]
dp[i + 1][k][1] |= dp[i][j][k]
else:
if j == k == a[i]:
pass
else:
dp[i + 1][k][a[i]] |= dp[i][j][k]
for a1 in range(2):
for a2 in range(2):
for b1 in range(2):
for b2 in range(2):
if not((a1 == a2 == b2) or (a1 == b1 == b2)) and dp[2][a1][a2] and dp[n][b1][b2]:
print('Yes')
return
print('No')
for _ in range(ii()):
solve()