結果
問題 | No.416 旅行会社 |
ユーザー | はむ吉🐹 |
提出日時 | 2016-08-27 19:55:46 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,654 ms / 4,000 ms |
コード長 | 2,392 bytes |
コンパイル時間 | 327 ms |
コンパイル使用メモリ | 87,340 KB |
実行使用メモリ | 219,504 KB |
最終ジャッジ日時 | 2023-08-21 09:55:02 |
合計ジャッジ時間 | 16,669 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 523 ms
158,236 KB |
testcase_01 | AC | 72 ms
71,668 KB |
testcase_02 | AC | 72 ms
71,600 KB |
testcase_03 | AC | 73 ms
71,672 KB |
testcase_04 | AC | 73 ms
71,420 KB |
testcase_05 | AC | 74 ms
71,564 KB |
testcase_06 | AC | 79 ms
75,260 KB |
testcase_07 | AC | 111 ms
77,608 KB |
testcase_08 | AC | 215 ms
81,816 KB |
testcase_09 | AC | 389 ms
89,852 KB |
testcase_10 | AC | 545 ms
158,212 KB |
testcase_11 | AC | 574 ms
166,724 KB |
testcase_12 | AC | 592 ms
166,252 KB |
testcase_13 | AC | 560 ms
166,384 KB |
testcase_14 | AC | 1,525 ms
218,128 KB |
testcase_15 | AC | 1,654 ms
219,504 KB |
testcase_16 | AC | 1,453 ms
217,036 KB |
testcase_17 | AC | 1,545 ms
217,076 KB |
testcase_18 | AC | 1,553 ms
218,648 KB |
testcase_19 | AC | 1,254 ms
189,020 KB |
testcase_20 | AC | 1,150 ms
185,200 KB |
ソースコード
#!/usr/bin/env pypy3 import array class UnionFind(object): def __init__(self, number_of_nodes): self.par = array.array("L", range(number_of_nodes)) self.rank = array.array("L", (0 for i in range(number_of_nodes))) self.group = [{i} for i in range(number_of_nodes)] def root(self, node): if self.par[node] == node: return node else: r = self.root(self.par[node]) self.par[node] = r return r def in_the_same_set(self, node1, node2): return self.root(node1) == self.root(node2) def unite(self, node1, node2): x = self.root(node1) y = self.root(node2) if x == y: pass elif self.rank[x] < self.rank[y]: self.par[x] = y self.group[y].update(self.group[x]) self.group[x].clear() else: self.par[y] = x self.group[x].update(self.group[y]) self.group[y].clear() if self.rank[x] == self.rank[y]: self.rank[x] += 1 def solve(n, q, all_edges, removed_edges): removed_edges_set = set(removed_edges) uf = UnionFind(n) con_time = array.array("l", (0 for _ in range(n))) root = 0 for edge in all_edges: if edge in removed_edges_set: continue uf.unite(*edge) for i in range(1, n): if uf.in_the_same_set(root, i): con_time[i] = -1 for t, (source, dest) in enumerate(removed_edges[::-1]): if not (uf.in_the_same_set(source, dest)): is_same_s = uf.in_the_same_set(root, source) is_same_d = uf.in_the_same_set(root, dest) if not (is_same_s or is_same_d): uf.unite(source, dest) else: if is_same_d: source, dest = dest, source for v in uf.group[uf.root(dest)]: con_time[v] = q - t uf.unite(source, dest) return con_time def main(): n, m, q = map(int, input().split()) all_edges = {tuple((int(x) - 1 for x in input().split())) for _ in range(m)} removed_edges = [tuple((int(x) - 1 for x in input().split())) for _ in range(q)] ans = solve(n, q, all_edges, removed_edges) print(*map(str, ans[1:]), sep="\n") if __name__ == '__main__': main()