結果
問題 | No.416 旅行会社 |
ユーザー |
![]() |
提出日時 | 2020-04-21 13:05:58 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,776 ms / 4,000 ms |
コード長 | 2,084 bytes |
コンパイル時間 | 220 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 129,912 KB |
最終ジャッジ日時 | 2024-12-14 20:50:12 |
合計ジャッジ時間 | 17,814 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 21 |
ソースコード
from itertools import permutationsimport syssys.setrecursionlimit(10 ** 6)from bisect import *from collections import *from heapq import *def II(): return int(sys.stdin.readline())def MI(): return map(int, sys.stdin.readline().split())def LI(): return list(map(int, sys.stdin.readline().split()))def SI(): return sys.stdin.readline()[:-1]def LLI(rows_number): return [LI() for _ in range(rows_number)]def LLI1(rows_number): return [LI1() for _ in range(rows_number)]int1 = lambda x: int(x) - 1def MI1(): return map(int1, sys.stdin.readline().split())def LI1(): return list(map(int1, sys.stdin.readline().split()))p2D = lambda x: print(*x, sep="\n")dij = [(1, 0), (0, 1), (-1, 0), (0, -1)]class UnionFind:def __init__(self, n):self.state = [-1] * nself.member = [[i] for i in range(n)]def root(self, u):v = self.state[u]if v < 0: return uself.state[u] = res = self.root(v)return resdef merge(self, u, v):res = []ru = self.root(u)rv = self.root(v)r0 = self.root(0)if ru == rv: return resif len(self.member[ru]) < len(self.member[rv]): ru, rv = rv, ruif ru == r0: res = self.member[rv]if rv == r0: res = self.member[ru]else:self.member[ru] += self.member[rv]self.member[rv] = []self.state[rv] = rureturn resdef main():# 橋を壊していくのではなく、クエリを逆順にして、橋をつないでいくn, m, q = MI()ab = LLI1(m)cd = LLI1(q)uf = UnionFind(n)# 破壊される橋をチェックdes = set()for c, d in cd: des.add((c, d))# 破壊されない橋を結ぶans = [0] * nfor a, b in ab:if (a, b) in des: continueuu = uf.merge(a, b)for u in uu: ans[u] = -1# クエリを逆から見て橋をつないでいくfor i, (c, d) in enumerate(cd[::-1]):uu = uf.merge(c, d)for u in uu: ans[u] = q - ifor i in range(1, n): print(ans[i])main()