CodingTest.백준 [트리의 부모찾기] python

문제 설명

트리의 순회

루트 없는 트리가 주어진다. 이때, 트리의 루트를 1이라고 정했을 때, 각 노드의 부모를 구하는 프로그램을 작성하시오.

코드 구현

아.. 너무 어렵다..

n=int(input())

tree=[[i] for i in range(n+1)]

for _ in range(n-1):
    a,b=list(map(int,input().split()))

    tree[a].append(b)
    tree[b].append(a)

print(tree)

q=[1]
answer = {}
result=[False for i in range(n+1)] 

print(result)

while len(q) > 0 :
  parent = q.pop(0)
  for i in tree[parent]:
    if not result[i]:
      answer[i] = parent
      q.append(i)
      result[i] = True
      print(answer)
      print(q)


print("\n".join(str(answer[i]) for i in range(2,n+1)))