글 읽기 - [Python3] C언어로 해결했던 문제인데 런타임 에러가 계속 발생합니다.

alsrl9 3년 전 0

동일한 로직으로 C언어를 이용해서 해결했던 문제입니다.

Python을 연습할 겸 Python으로도 작성했는데 번번히 런타임 에러가 발생하네요.

만 하루를 꼬박 고민해도 어떤 점이 런타임 에러를 일으킬 수 있는지 감도 안 잡힙니다.

도와주시면 감사하겠습니다.

import sys
from math import log2, ceil

N = int(sys.stdin.readline())
conn = [[] for i in range(N+1)] # connection information
for i in range(N-1):
    node1, node2 = map(int, sys.stdin.readline().split())
    conn[node1].append(node2)
    conn[node2].append(node1)

LOG = ceil(log2(N-1))
parent = [[0 for i in range(LOG+1)] for j in range(N+1)]
depth = [0 for i in range(N+1)]
check = [False for i in range(N+1)]

def dfs(node:int, d:int):
    global parent, depth, check
    depth[node] = d
    check[node] = True
    for i in range(len(conn[node])):
        nextNode = conn[node][i]
        if check[nextNode]:
            continue
        parent[nextNode][0] = node
        dfs(nextNode, d+1)

dfs(1, 0)
for i in range(1, LOG+1):
    for j in range(1, N+1):
        parent[j][i] = parent[parent[j][i-1]][i-1]

M = int(sys.stdin.readline())
for i in range(M):
    node1, node2 = map(int, sys.stdin.readline().split())
    if depth[node1] > depth[node2]:
        node1, node2 = node2, node1
    
    for i in range(LOG, -1, -1):
        if depth[node2] - depth[node1] >= (1 << i):
            node2 = parent[node2][i]
    
    if node1 == node2:
        print(node1)
        continue

    for i in range(LOG, -1, -1):
        if parent[node1][i] != parent[node2][i]:
            node1 = parent[node1][i]
            node2 = parent[node2][i]
    print(parent[node1][0])

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

[Python3] C언어로 해결했던 문제인데 런타임 에러가 계속 발생합니다.

alsrl9 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말