結果

問題 No.416 旅行会社
ユーザー nebukuro09nebukuro09
提出日時 2016-10-19 17:20:01
言語 PyPy2
(7.3.13)
結果
OLE  
実行時間 -
コード長 1,672 bytes
コンパイル時間 1,596 ms
コンパイル使用メモリ 77,492 KB
実行使用メモリ 324,900 KB
最終ジャッジ日時 2023-08-14 17:40:45
合計ジャッジ時間 31,358 ms
ジャッジサーバーID
(参考情報)
judge12 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 OLE -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 OLE -
testcase_09 OLE -
testcase_10 OLE -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 OLE -
testcase_15 OLE -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import copy

class UnionFind(object):
    def __init__(self, n):
        self.uf = [-1 for _ in xrange(n)]
        self.sets = {0:set([0])}

    def union(self, x, y):
        x = self.find(x)
        y = self.find(y)
        if x == y:
            return []
        if self.uf[x] >= self.uf[y]:
            if self.uf[x] == self.uf[y] == -1:
                self.sets[y] = set([x, y])
            elif self.uf[x] == -1:
                self.sets[y].add(x)
            else:
                self.sets[y] |= self.sets[x]
            self.uf[y] += self.uf[x]
            self.uf[x] = y
            return self.sets[y]
        else:
            if self.uf[y] == -1:
                self.sets[x].add(y)
            else:
                self.sets[x] |= self.sets[y]
            self.uf[x] += self.uf[y]
            self.uf[y] = x
            return self.sets[x]

    def find(self, x):
        leaves = []
        while self.uf[x] >= 0:
            leaves.append(x)
            x = self.uf[x]
        for lf in leaves:
            self.uf[lf] = x
        return x
    
N, M, Q = map(int, raw_input().split())
AB = set([tuple(map(lambda x:int(x)-1, raw_input().split())) for _ in xrange(M)])
CD = [tuple(map(lambda x:int(x)-1, raw_input().split())) for _ in xrange(Q)]
AB -= set(CD)

uf = UnionFind(N)
for a, b in AB:
    uf.union(a, b)
ans = [-1 for _ in xrange(N)]
old_set = set(uf.sets[uf.find(0)])
print old_set
for i in xrange(Q-1, -1, -1):
    c, d = CD[i]
    uf.union(c, d)
    new_set = uf.sets[uf.find(0)]
    print i, new_set, old_set
    for j in new_set - old_set:
        ans[j] = i+1
    old_set = copy.copy(new_set)
for i in xrange(1, N):
    print ans[i]
    
0