restructure zeta/** to storage/zeta/**

This commit is contained in:
2025-05-10 21:54:24 +09:00
parent 2886820691
commit 2f2e0759fd
407 changed files with 7 additions and 1 deletions

View File

@@ -0,0 +1,32 @@
import sys
input = sys.stdin.readline
print = sys.stdout.write
def root(P, e):
node = e
while node != P[node]:
node = P[node]
return node
if __name__ == "__main__":
N, M = map(int, input().split())
P = [i for i in range(N + 1)]
for _ in range(M):
op, a, b = map(int, input().split())
if op == 0: # Merge
if a == b:
continue
rb, ra = root(P, b), root(P, a)
if ra < rb:
P[rb] = ra
elif ra > rb:
P[ra] = rb
elif op == 1: # Find
print("YES" if a == b or root(P, a) == root(P, b) else "NO")
print("\n")