結果

問題 No.1078 I love Matrix Construction
ユーザー vwxyzvwxyz
提出日時 2023-05-17 10:27:00
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,898 bytes
コンパイル時間 216 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 213,760 KB
最終ジャッジ日時 2024-05-08 21:59:56
合計ジャッジ時間 11,263 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
11,136 KB
testcase_01 AC 33 ms
11,136 KB
testcase_02 AC 344 ms
40,704 KB
testcase_03 AC 914 ms
88,576 KB
testcase_04 AC 1,369 ms
118,400 KB
testcase_05 AC 1,268 ms
100,480 KB
testcase_06 AC 323 ms
39,296 KB
testcase_07 AC 130 ms
21,760 KB
testcase_08 AC 1,080 ms
99,712 KB
testcase_09 AC 72 ms
15,360 KB
testcase_10 TLE -
testcase_11 AC 1,444 ms
124,288 KB
testcase_12 TLE -
testcase_13 TLE -
testcase_14 AC 1,731 ms
141,696 KB
testcase_15 TLE -
testcase_16 AC 113 ms
19,584 KB
testcase_17 AC 32 ms
11,136 KB
testcase_18 AC 253 ms
32,768 KB
testcase_19 AC 571 ms
60,672 KB
testcase_20 AC 557 ms
59,776 KB
testcase_21 AC 45 ms
12,672 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
readline=sys.stdin.readline

def SCC(N,edges):
    start = [0] * (N + 1)
    elist = [0] * len(edges)
    for e in edges:
        start[e[0] + 1] += 1
    for i in range(1, N + 1):
        start[i] += start[i - 1]
    counter = start[:]
    for e in edges:
        elist[counter[e[0]]] = e[1]
        counter[e[0]] += 1
    N = N
    now_ord = group_num = 0
    visited = []
    low = [0] * N
    order = [-1] * N
    ids = [0] * N
    parent = [-1] * N
    stack = []
    for i in range(N):
        if order[i] == -1:
            stack.append(i)
            stack.append(i)
            while stack:
                v = stack.pop()
                if order[v] == -1:
                    low[v] = order[v] = now_ord
                    now_ord += 1
                    visited.append(v)
                    for i in range(start[v], start[v + 1]):
                        to = elist[i]
                        if order[to] == -1:
                            stack.append(to)
                            stack.append(to)
                            parent[to] = v
                        else:
                            low[v] = min(low[v], order[to])
                else:
                    if low[v] == order[v]:
                        while True:
                            u = visited.pop()
                            order[u] = N
                            ids[u] = group_num
                            if u == v:
                                break
                        group_num += 1
                    if parent[v] != -1:
                        low[parent[v]] = min(low[parent[v]], low[v])
    for i, x in enumerate(ids):
        ids[i] = group_num - 1 - x
    groups = [[] for _ in range(group_num)]
    for i, x in enumerate(ids):
        groups[x].append(i)
    return groups

class TwoSAT:
    def __init__(self,N):
        self.N=N
        self.edges=[]

    def Add_Clause(self,i,f,j,g):
        assert 0<=i<self.N
        assert 0<=j<self.N
        self.edges.append((2*i+(0 if f else 1),2*j+(1 if g else 0)))
        self.edges.append((2*j+(0 if g else 1),2*i+(1 if f else 0)))

    def Satisfiable(self):
        scc=SCC(2*self.N,self.edges)
        idx=[None]*2*self.N
        for i,lst in enumerate(scc):
            for x in lst:
                idx[x]=i
        retu=[None]*self.N
        for i in range(self.N):
            if idx[2*i]==idx[2*i+1]:
                return None
            retu[i]=idx[2*i]<idx[2*i+1]
        return retu

N=int(readline())
S=list(map(int,readline().split()))
T=list(map(int,readline().split()))
U=list(map(int,readline().split()))
TSAT=TwoSAT(N**2)
for s,t,u in zip(S,T,U):
    s-=1;t-=1
    for j in range(N):
        TSAT.Add_Clause(s*N+j,u&1,j*N+t,u&2)
lst=TSAT.Satisfiable()
if lst:
    ans_lst=[[int(lst[i*N+j])^1 for j in range(N)]for i in range(N)]
    for ans in ans_lst:
        print(*ans)
else:
    print(-1)
0