[백준] 7511번 소셜 네트워킹 어플리케이션 파이썬 코드

코드

import sys
sys.setrecursionlimit(10**5)
input = sys.stdin.readline

def find(a, lst):
  if a != lst[a]:
    lst[a] = find(lst[a], lst)
  return lst[a]

def union(a, b, lst):
  a = find(a,lst)
  b = find(b,lst)
  
  if a < b:
    lst[b] = a
  else: lst [a] = b
tc = int(input())
for i in range(tc):
  n = int(input())
  k = int(input())
  parent = [i for i in range(n+1)]
  for _ in range(k):
    a,b = map(int, input().split())
    union(a,b,parent)
    
  m = int(input())
  print(f'Scenario {i+1}:')
  for _ in range(m):
    u, v = map(int, input().split())
    if find(u, parent) == find(v, parent):
      print(1)
    else: print(0)
  print()

입력 조건 해석에 시간이 좀 걸렸던 문제. 여러 개의 테스트 케이스라고 해서 무한 루프인가 했는데 테스트 케이스 개수가 주어진 것이었다. 그 이외에는 유니온 파인드 구현만으로 해결할 수 있는 문제.