結果
問題 | No.416 旅行会社 |
ユーザー | qib |
提出日時 | 2022-11-15 22:13:41 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,662 bytes |
コンパイル時間 | 247 ms |
コンパイル使用メモリ | 87,008 KB |
実行使用メモリ | 199,724 KB |
最終ジャッジ日時 | 2023-10-15 03:24:56 |
合計ジャッジ時間 | 12,019 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 360 ms
136,700 KB |
testcase_01 | AC | 68 ms
71,112 KB |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | AC | 368 ms
136,624 KB |
testcase_11 | AC | 375 ms
136,672 KB |
testcase_12 | AC | 400 ms
136,684 KB |
testcase_13 | AC | 342 ms
136,692 KB |
testcase_14 | AC | 814 ms
199,500 KB |
testcase_15 | AC | 863 ms
199,404 KB |
testcase_16 | AC | 833 ms
199,596 KB |
testcase_17 | AC | 812 ms
199,724 KB |
testcase_18 | AC | 796 ms
199,492 KB |
testcase_19 | WA | - |
testcase_20 | WA | - |
ソースコード
class UnionFind: def __init__(self, n): self.node = [-1 for _ in range(n)] def root(self, v): if self.node[v] < 0: return v st = [] while self.node[v] >= 0: st.append(v) v = self.node[v] for u in st: self.node[u] = v return v def size(self, v): v = self.root(v) return (- self.node[v]) def same(self, u, v): return self.root(u) == self.root(v) def unite(self, u, v): ru = self.root(u) rv = self.root(v) if ru == rv: return du = self.node[ru] dv = self.node[rv] if du <= dv: self.node[rv] = ru self.node[ru] += dv else: self.node[ru] = rv self.node[rv] += du n, m, q = map(int, input().split()) edges = [] idx = {} for _ in range(m): a, b = map(int, input().split()) a -= 1 b -= 1 edges.append((a, b)) idx[a, b] = len(idx) queries = [] for _ in range(q): c, d = map(int, input().split()) c -= 1 d -= 1 queries.append(idx[c, d]) es = set([i for i in range(m)]) - set(queries) uf = UnionFind(n) comps = [[] for _ in range(n)] for e in es: a, b = edges[e] uf.unite(a, b) for v in range(n): r = uf.root(v) comps[r].append(v) ans = [None for _ in range(n)] for v in comps[0]: ans[v] = -1 for i in range(q - 1, -1, -1): c, d = edges[queries[i]] rc = uf.root(c) rd = uf.root(d) if rc == rd: continue r0 = uf.root(0) if r0 == rc: for v in comps[rd]: ans[v] = i + 1 elif r0 == rd: for v in comps[rc]: ans[v] = i + 1 uf.unite(rc, rd) r = uf.root(rc) while len(comps[rc + rd - r]) > 0: comps[r].append(comps[rc + rd - r].pop()) print(*ans[1:], sep="\n")