[백준] 11725번 트리의 부모 찾기 파이썬 코드

코드

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

a = int(input())
tree = [[] for _ in range(a+1)]
visited = [False] * (a+1) # for 방문여부 체크
parents = [0] * (a+1) # for 부모노드 기록

# 그래프 작성

for _ in range(a-1):
  x,y = map(int, input().split())
  tree[x].append(y)
  tree[y].append(x)

def dfs(v, graph, visit_check):
  visit_check[v] = True    # 방문 여부 확인
  for i in graph[v]:
    if visit_check[i] != True:
      parents[i] = v
      dfs(i, graph, visit_check)
  return 0    # 임의의 함수 반환값. 확인할 것은 parent이므로 반환값은 중요하지 않다.
 
dfs(1, tree, visited)

for i in range(2, a+1):
  print(parents[i])

10/27 재시도 코드

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**6)
node = int(input())
graph =[[] for _ in range(node+1)]
visited = [0] * (node + 1)
for _ in range(node-1):
  x, y = map(int, input().split())
  graph[x].append(y)
  graph[y].append(x)

def dfs(i):  
  for k in graph[i]:
    if not visited[k]:
      visited[k] = i
      dfs(k)
dfs(1)
for i in range(2, node+1):
  print(visited[i])