結果

問題 No.1078 I love Matrix Construction
ユーザー ttrttr
提出日時 2020-06-13 19:37:51
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,047 ms / 2,000 ms
コード長 2,432 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 234,032 KB
最終ジャッジ日時 2024-06-25 19:21:06
合計ジャッジ時間 12,997 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 49 ms
54,528 KB
testcase_01 AC 48 ms
54,144 KB
testcase_02 AC 256 ms
97,548 KB
testcase_03 AC 482 ms
139,852 KB
testcase_04 AC 611 ms
162,424 KB
testcase_05 AC 543 ms
153,736 KB
testcase_06 AC 260 ms
96,412 KB
testcase_07 AC 186 ms
84,296 KB
testcase_08 AC 528 ms
149,592 KB
testcase_09 AC 154 ms
80,296 KB
testcase_10 AC 1,047 ms
234,032 KB
testcase_11 AC 622 ms
159,836 KB
testcase_12 AC 879 ms
213,016 KB
testcase_13 AC 968 ms
222,208 KB
testcase_14 AC 697 ms
173,492 KB
testcase_15 AC 955 ms
217,012 KB
testcase_16 AC 184 ms
83,600 KB
testcase_17 AC 46 ms
54,144 KB
testcase_18 AC 222 ms
90,500 KB
testcase_19 AC 337 ms
109,728 KB
testcase_20 AC 345 ms
109,128 KB
testcase_21 AC 116 ms
77,824 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
N = int(input())
S = [int(s)-1 for s in input().split()]
T = [int(t)-1 for t in input().split()]
U = [int(u) for u in input().split()]

M = N**2*2
E = [[] for _ in range(M)]
E2 = [[] for _ in range(M)]
used = [0]*M

for i in range(N):
    for j in range(N):
        l0 = (S[i]*N+j)*2
        l1 = l0+1
        r0 = (j*N+T[i])*2
        r1 = r0+1
        if U[i] == 0:
            E[l0].append(r1)
            E[r0].append(l1)
            E2[l1].append(r0)
            E2[r1].append(l0)
        elif U[i] == 1:
            E[l1].append(r1)
            E[r0].append(l0)
            E2[l0].append(r0)
            E2[r1].append(l1)
        elif U[i] == 2:
            E2[l1].append(r1)
            E2[r0].append(l0)
            E[l0].append(r0)
            E[r1].append(l1)
        else:
            E2[l0].append(r1)
            E2[r0].append(l1)
            E[l1].append(r0)
            E[r1].append(l0)

L = []
for i in range(N):
    for j in range(N):
        for k in range(2):
            n = (i*N+j)*2+k
            if used[n] == 1:
                continue
            q = deque()
            q.append(n)
            used[n] = 1
            h = deque()
            h.append(0)
            while q:
                temp = q[-1]
                if h[-1] == len(E[temp]):
                    q.pop()
                    h.pop()
                    L.append(temp)
                else:
                    j2 = E[temp][h[-1]]
                    h[-1] += 1
                    if used[j2] == 0:
                        used[j2] = 1
                        q.append(j2)
                        h.append(0)

A = [[-1]*N for _ in range(N)]
L2 = []
L.reverse()
q = deque()
for l in L:
    if used[l] == 2:
        continue
    ele = []
    used[l] = 2
    q.append(l)
    while q:
        temp = q.pop()
        ele.append(temp)
        for e in E2[temp]:
            if used[e] == 2:
                continue
            used[e] = 2
            q.append(e)
    L2.append(ele)

L2.reverse()
used = [-1]*M
for i in range(len(L2)):
    for l in L2[i]:
        used[l] = i

for i in range(N):
    for j in range(N):
        n = (i*N+j)*2
        if used[n] > used[n+1]:
            A[i][j] = 1
        elif used[n] == used[n+1]:
            A = -1
            break
        else:
            A[i][j] = 0
    if A == -1:
        break

if A == -1:
    print(A)
else:
    for i in range(N):
        print(*A[i])
0