結果

問題 No.1078 I love Matrix Construction
コンテスト
ユーザー QCFium
提出日時 2020-05-02 18:20:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,173 ms / 2,000 ms
コード長 1,839 bytes
コンパイル時間 946 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 185,484 KB
最終ジャッジ日時 2024-12-30 20:35:48
合計ジャッジ時間 17,039 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(500 * 500 * 2 + 1)

N = int(input())
a = list(map(int, input().split()))
b = list(map(int, input().split()))
c = list(map(int, input().split()))
for i in range(N) :
	a[i] -= 1
	b[i] -= 1

def INDEX(i, j, k) : return (i * N + j) * 2 + k
hen = [[] for i in range(N * N * 2)]
rev = [[] for i in range(N * N * 2)]
for i in range(N) :
	r0 = c[i] % 2
	r1 = c[i] // 2
	for j in range(N) :
		hen[INDEX(a[i], j, r0)].append(INDEX(j, b[i], not r1));
		rev[INDEX(j, b[i], not r1)].append(INDEX(a[i], j, r0));
		hen[INDEX(j, b[i], r1)].append(INDEX(a[i], j, not r0));
		rev[INDEX(a[i], j, not r0)].append(INDEX(j, b[i], r1));

visited = [False for i in range(N * N * 2)]
ord = []

def dfs0(i) :
	visited[i] = True
	stack = [i]
	head = [0]
	while len(stack) :
		i = stack[-1]
		if head[-1] == len(hen[i]) :
			stack.pop(-1)
			head.pop(-1)
			ord.append(i)
		else :
			j = hen[i][head[-1]]
			head[-1] += 1
			if not visited[j] :
				visited[j] = True
				stack.append(j)
				head.append(0)

group = [-1 for i in range(N * N * 2)]
group_cnt = 0

def dfs1(i) :
	group[i] = group_cnt
	stack = [i]
	head = [0]
	while len(stack) :
		i = stack[-1]
		if head[-1] == len(rev[i]) :
			stack.pop(-1)
			head.pop(-1)
		else :
			j = rev[i][head[-1]]
			head[-1] += 1
			if group[j] == -1 :
				stack.append(j)
				head.append(0)
				group[j] = group_cnt

for i in range(N * N * 2) :
	if not visited[i] : dfs0(i)
ord = ord[::-1]
for i in ord :
	if group[i] == -1 :
		dfs1(i)
		group_cnt += 1

possible = True
res = [[0 for j in range(N)] for i in range(N)]
for i in range(N) :
	for j in range(N) :
		if group[INDEX(i, j, 0)] == group[INDEX(i, j, 1)] : possible = False
		res[i][j] = int(group[INDEX(i, j, 0)] < group[INDEX(i, j, 1)])

if not possible :
	print("-1")
else :
	for i in range(N) :
		print(' '.join(map(str, res[i])))
0