Submission #912827

---

Source Code Expand

import array
 
def read():
    return int(input())
 
def reads(sep=None):
    return list(map(int, input().split(sep)))
 
def root(table, x):
    if table[x] == x:
        return x
    else:
        table[x] = root(table, table[x])
        return table[x]
 
def main():
    n, q = reads()
    
    table = array.array('I', [i for i in range(n)])
    
    for _ in range(q):
        p, a, b = reads()
        if p == 0:
            rx = root(table, a)
            ry = root(table, b)

            if rx < ry:
                table[ry] = rx
            else:
                table[rx] = ry
        else:
            if root(table, a) == root(table, b):
                print('Yes')
            else:
                print('No')
    
main()

Submission Info

Judge Result