結果
問題 | No.1078 I love Matrix Construction |
ユーザー | roaris |
提出日時 | 2020-12-24 21:28:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,049 ms / 2,000 ms |
コード長 | 2,945 bytes |
コンパイル時間 | 888 ms |
コンパイル使用メモリ | 81,836 KB |
実行使用メモリ | 226,680 KB |
最終ジャッジ日時 | 2023-10-21 15:50:48 |
合計ジャッジ時間 | 13,237 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 32 ms
53,344 KB |
testcase_01 | AC | 34 ms
53,348 KB |
testcase_02 | AC | 255 ms
93,576 KB |
testcase_03 | AC | 459 ms
123,540 KB |
testcase_04 | AC | 633 ms
146,708 KB |
testcase_05 | AC | 530 ms
127,516 KB |
testcase_06 | AC | 196 ms
92,868 KB |
testcase_07 | AC | 179 ms
83,700 KB |
testcase_08 | AC | 594 ms
141,244 KB |
testcase_09 | AC | 116 ms
79,248 KB |
testcase_10 | AC | 1,049 ms
226,680 KB |
testcase_11 | AC | 647 ms
149,460 KB |
testcase_12 | AC | 841 ms
184,204 KB |
testcase_13 | AC | 942 ms
207,672 KB |
testcase_14 | AC | 724 ms
161,832 KB |
testcase_15 | AC | 1,021 ms
196,244 KB |
testcase_16 | AC | 165 ms
81,620 KB |
testcase_17 | AC | 33 ms
53,344 KB |
testcase_18 | AC | 231 ms
89,132 KB |
testcase_19 | AC | 269 ms
104,784 KB |
testcase_20 | AC | 310 ms
105,144 KB |
testcase_21 | AC | 72 ms
76,552 KB |
ソースコード
import sys input = sys.stdin.readline sys.setrecursionlimit(250010) class Scc: def __init__(self, N, G): self.N = N self.G = G self.RG = [[] for _ in range(N)] for v in range(N): for nv in G[v]: self.RG[nv].append(v) def decomp(self): order = [] visited = [False]*self.N def dfs(v): visited[v] = True for nv in self.G[v]: if not visited[nv]: dfs(nv) order.append(v) for v in range(self.N): if not visited[v]: dfs(v) visited = [False]*self.N comp = [-1]*self.N label = 0 def rdfs(v, label): comp[v] = label visited[v] = True for nv in self.RG[v]: if not visited[nv]: rdfs(nv, label) for v in reversed(order): if not visited[v]: rdfs(v, label) label += 1 return label, comp def construct(self): label, comp = self.decomp() belong = [[] for _ in range(label)] nG = [set() for _ in range(label)] for v in range(self.N): for nv in self.G[v]: if comp[v]!=comp[nv]: nG[comp[v]].add(comp[nv]) belong[comp[v]].append(v) return belong, nG class Two_sat: def __init__(self, N): self.N = N self.G = [[] for _ in range(2*N)] def add_clause(self, i, f, j, g): self.G[2*i+(0 if f else 1)].append(2*j+(1 if g else 0)) self.G[2*j+(0 if g else 1)].append(2*i+(1 if f else 0)) def allocate(self): scc = Scc(2*self.N, self.G) label, comp = scc.decomp() res = [-1]*self.N for i in range(self.N): if comp[2*i]==comp[2*i+1]: return -1 res[i] = 1 if comp[2*i]<comp[2*i+1] else 0 return res N = int(input()) S = list(map(int, input().split())) T = list(map(int, input().split())) U = list(map(int, input().split())) ts = Two_sat(N*N) for Si, Ti, Ui in zip(S, T, U): if Ui==0: for j in range(N): ts.add_clause((Si-1)*N+j, True, j*N+Ti-1, True) elif Ui==1: for j in range(N): ts.add_clause((Si-1)*N+j, False, j*N+Ti-1, True) elif Ui==2: for j in range(N): ts.add_clause((Si-1)*N+j, True, j*N+Ti-1, False) else: for j in range(N): ts.add_clause((Si-1)*N+j, False, j*N+Ti-1, False) res = ts.allocate() if res==-1: print(-1) exit() ans = [[0]*N for _ in range(N)] for i in range(N): for j in range(N): ans[i][j] = res[i*N+j] for ans_i in ans: print(*ans_i)