結果

問題 No.2518 Adjacent Larger
コンテスト
ユーザー KA37RI
提出日時 2023-10-28 00:25:53
言語 Python3
(3.14.2 + numpy 2.4.0 + scipy 1.16.3)
結果
WA  
実行時間 -
コード長 913 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 363 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 70,692 KB
最終ジャッジ日時 2024-09-25 15:46:18
合計ジャッジ時間 8,087 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other AC * 3 WA * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys

def solve(n, a):
	larger = [set() for _ in range(n)]
	for i in range(n):
		if a[i] == 0:
			if (i - 1) % n in larger[i]:
				print("contradiction", file=sys.stderr)
				return False
			if (i + 1) % n in larger[i]:
				print("contradiction", file=sys.stderr)
				return False
			larger[(i-1)%n].add(i)
			larger[(i+1)%n].add(i)
		elif a[i] == 2:
			if i in larger[(i-1)%n]:
				print("contradiction", file=sys.stderr)
				return False
			if i in larger[(i+1)%n]:
				print("contradiction", file=sys.stderr)
				return False
			larger[i].add((i - 1) % n)
			larger[i].add((i + 1) % n)
	for i in range(n):
		if a[i] == 1:
			if (i - 1) % n in larger[i]:
				larger[(i+1)%n].add(i)
			if (i + 1) % n in larger[i]:
				larger[(i-1)%n].add(i)
	print(larger, file=sys.stderr)

t = int(input())
for _ in range(t):
	n = int(input())
	a = [int(x) for x in input().split()]
	print("Yes" if solve(n, a) else "No")
0