15-pu2rile

[pu2rile]

🔗 문제 링크

백준 - 나무 탈출

✔️ 소요된 시간

1시간

✨ 수도 코드

문제 요약

나무 탈출 게임

이 게임은 두 사람이 번갈아 가면서 게임판에 놓여있는 게임말을 움직이는 게임입니다. 처음에는 트리의 모든 리프 노드에 게임말이 하나씩 놓여있는 채로 시작합니다. 어떤 사람의 차례가 오면, 그 사람은 현재 존재하는 게임말 중 아무거나 하나를 골라 그 말이 놓여 있던 노드의 부모 노드로 옮깁니다. 이렇게 옮긴 후에 만약 그 게임말이 루트 노드에 도착했다면 그 게임말을 제거합니다. 모든 과정을 마치면 다음 사람에게 차례를 넘기고, 게임말이 게임판에 존재하지 않아 고를 수 없는 사람이 지게 됩니다.

문제 목표

주어진 트리에서 모든 리프 노드가 루트로 돌아가는 데 필요한 총 이동 횟수를 계산한 후, 그 값이 홀수인지 짝수인지 판단하여 첫 번째 플레이어가 이길지 두 번째 플레이어가 이길지를 결정합니다.

문제 풀이

리프 노드의 깊이를 구한 후, 리프 노드들의 깊이의 합이 홀수이면 Yes(성원이가 이김), 짝수이면 No(성원이가 짐)가 출력되도록 만들면 됩니다.

함수 설명

def dfs(cur, prv, dep):
    arr[cur] = dep
    for next in tree[cur]:
        if next == prv:
            continue
        dfs(next, cur, dep+1)

깊이 우선 탐색(DFS)을 사용하여 트리 구조를 탐색하고, 각 노드까지의 깊이를 arr 리스트에 저장하는 함수입니다.

• 매개변수
  • cur : 현재 탐색 중인 노드 번호
  • prv : 이전에 방문한 노드 번호(= 부모 노드)
  • dep : 현재 노드까지의 깊이
• arr[cur] = dep : 현재 노드(cur)까지의 거리(깊이)를 arr 리스트의 cur 인덱스에 저장
• 재귀 호출
  • dfs(next, cur, dep + 1) : 재귀적으로 next 노드에 대해 깊이 우선 탐색을 수행
  • cur 이 next 노드의 부모가 되고, dep + 1 을 통해 현재 깊이를 1 증가

최종 코드

보기

import sys
sys.setrecursionlimit(10 ** 6)

n = int(sys.stdin.readline().strip())
# 트리에 저장
tree = [[] for _ in range (n+1)]
for _ in range(n-1):
    a,b = map(int, sys.stdin.readline().strip().split())
    tree[a].append(b)
    tree[b].append(a)

arr = [0] * (n + 1)
def dfs(cur, prv, dep):
    arr[cur] = dep  # cnt는 현재 노드(cur)까지의 거리를 arr 리스트의 cur 인덱스에 저장
    for next in tree[cur]:  # 현재 노드(cur)에 연결된 모든 자식 노드를 순회
        if next == prv: # 다음 노드(next)가 이전에 방문한 부모 노드(prv)라면
            continue    # 건너뛰기
        dfs(next,cur,dep+1)  # 재귀적으로 dfs 수행, cur이 next 노드의 부모가 되고, cnt + 1을 통해 현재 깊이를 1 증가

dfs(1, 0, 0)    # 트리 탐색 시작
cnt = 0
for i in range(2, n+1):
    # 리프 노드의 깊이 더하기
    if len(tree[i]) == 1:
        cnt += arr[i]

if cnt % 2 == 1:
    print("Yes")
else:
    print("No")

📚 새롭게 알게된 내용

처음 코드를 제출했을 때 런타임 에러가 떠서 왜 이러지 하고 검색해 보니 파이썬의 기본 재귀 깊이 제한 때문이라고 하더라고요!
재귀를 이용한 풀이에서는 이런 런타임 에러를 방지하기 위해 sys.setrecursionlimit(10 ** 6) 와 같은 코드를 추가해야 한다는 것을 알게 되었습니다~