Submission #4262698

---

Source Code Expand

class UnionFind():
    n=1
    par=[0]
    rnk=[0]
    def __init__(self,size):
        self.n=size
        self.par=[i for i in range(self.n)]
        self.rnk=[0 for i in range(self.n)]
    def find(self,vertex1):
        if self.par[vertex1]==vertex1:
            return vertex1
        else:
            self.par[vertex1]=self.find(self.par[vertex1])
            return self.par[vertex1]
    def unite(self,vertex1,vertex2):
        vertex1=self.find(vertex1)
        vertex2=self.find(vertex2)
        if vertex1==vertex2:
            return
        if (self.rnk[vertex1]<self.rnk[vertex2]):
            self.par[vertex1]=vertex2
        else:
            self.par[vertex2]=vertex1
            if(self.rnk[vertex1]==self.rnk[vertex2]):
                self.rnk[vertex1]+=1
    def same(self,vetrex1,vertex2):
        return self.find(vetrex1)==self.find(vertex2)

#N,Qを標準入力から受け取る
N,Q=map(int,input().split())
#UnionFindクラスの初期化,最初の頂点数はNとしている。
G=UnionFind(N)
for i in range(Q):
    #クエリの入力を受け取る
    p,a,b=map(int,input().split())
    #入力では1-indexedなため処理用に0-indexedに変更する
    if p==0:
        #連結クエリ
        #aとbを繋げる
        G.unite(a,b)
    else:
        #判定クエリ
        #G.same(a,b)は繋がっていたらTrue,繋がってないならFalseを返す
        if G.same(a,b):
            print("Yes")
        else:
            print("No")

Submission Info

Judge Result