공통 조상 찾기

문제

트리의 노드 X에 대하여 “조상"을 정의할 수 있다. X의 “조상"이란, 루트까지 올라가는 중에 만나는 모든 노드를 말한다. 예를 들어, 아래와 같이 트리가 주어질 경우, 노드 8의 “조상"은 노드 0, 노드 2, 노드 6이 된다.

두 노드 X, Y의 공통 조상이란, X와 Y가 공통으로 갖는 조상을 말한다. 예를 들어, 노드 7과 노드 10의 공통조상은 노드 2, 노드 0이 된다. 가장 가까운 공통 조상이란, X와 Y가 공통으로 갖는 조상들 중에서 X, Y와 가장 가까운 조상을 말한다. 예를 들어, 노드 7과 노드 10의 가장 가까운 공통 조상은 노드 2가 된다. 트리가 주어지고, 두 노드 X, Y가 주어질 때, 가장 가까운 공통 조상을 찾는 프로그램을 작성하시오.

 

입력

첫 번째 줄에 트리의 노드 개수 n, 두 노드 X, Y의 번호가 주어진다. ( 1 ≤ X, Y ≤ n ≤ 1000 ) 두 번째 줄부터 트리의 간선 정보가 주어진다. 각 줄은 2개의 숫자 a, b로 이루어지며, 이는 노드 a가 노드 b의 부모노드라는 것을 의미한다. 루트는 노드 0이라고 가정한다.  

출력

두 노드 X, Y의 공통 조상을 출력한다.

 

예제 입력

11 7 10

0 1

0 2

1 3

1 4

1 5

2 6

2 10

6 7

6 8

6 9

예제 출력

2

 

 

#코드

## template
import sys
n, x, y = map(int, sys.stdin.readline().split())
getInfo = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
frame = {}
x_list = list()
y_list = list()
 
for item in getInfo:
  if len(item) > 0: 
    try: frame[item[1]] = item[0]
    except: frame[item[1]] = item[0]
 
def find_x_p(x):
  if x == 0:
    return
  else:
    for k, v in frame.items():
      if x == k:
        x_list.append(v)
        find_x_p(v)
        
def find_y_p(y):
  if y == 0:
    return
  else:
    for k, v in frame.items():
      if y == k:
        y_list.append(v)
        find_y_p(v)
 
find_x_p(x) 
find_y_p(y) 
flag = True
for i in range(len(x_list)):
  for j in range(len(y_list)):
    if flag == True and x_list[i] == y_list[j]:
      print(x_list[i])
      flag = False
      break