[boj] 백준 25601 자바의 형변환 python 풀이

dictionary를 이용해 단방향 링크드리스트 구조로 구현.

부모의 자식은 여러 명일 수도 있지만, 자식의 부모는 무조건 1명이라는 점을 이용해 순회함.

 

문제)

https://www.acmicpc.net/problem/25601

 

코드)

 

import sys
input = sys.stdin.readline

N = int(input().strip())

tree = dict()

for i in range(N-1):
    A, B = input().split() # A: 자식, B: 부모 
    tree[A] = B # 부모는 언제나 1명

A, B = input().split()

def check(tree, child, target_parent):
    parent = child
    while True:
        try:
            parent = tree[parent]
            # print(parent)

            if parent == target_parent:
                return 1

        except:
            return 0 

print(check(tree, A, B) | check(tree, B, A))