글 읽기 - [파이썬] sys.stdin 런타임에러

ponta12 6년 전 0

input 써서 하다가 시간초과 떠서 이 부분을 sys.stdin으로 바꿔 줬는데 런타임 에러가 나내요... 이유가 뭘까요...ㅠㅠ

import sys
V, E = map(int, input().split())
varr = [x for x in range(V + 1)]
arr = []
for y in range(E):
    temp = [int(x) for x in sys.stdin.readline()[:-1].split()]
    temp.reverse()
    arr.append(temp)
print(arr)
arr.sort()
Sum = 0
count = 0
for x in range(E):
    a = arr[x][2]
    b = arr[x][1]
    while a != varr[a] or b != varr[b]:
        a = varr[a]
        b = varr[b]
    if a != b:
        for y in range(1, V + 1):
            if varr[y] == b:
                varr[y] = a
        varr[arr[x][1]] = a
        Sum += arr[x][0]
        count += 1
        if count == V-1:
            break
    else:
        continue
print(Sum)

jh05013 6년 전 1

입력의 끝이 꼭 개행문자일 필요는 없습니다. [:-1] 없이 바로 split을 해도 맨 끝의 공백을 무시하고 나눌 수 있습니다.

ponta12 6년 전 0

jh05013 -1 하나 차이로 런타임 에러 발생이 되지 않는 이유가 뭔지 궁금합니다....

jh05013 6년 전 1

개행문자가 있으면 "1 2 3\n"에 [:-1]을 했을 때 "1 2 3"이 되고, 이것을 split해서 리스트로 저장할 수 있습니다. 하지만 개행문자가 없으면 "1 2 3"에 [:-1]을 해서 "1 2 "가 되므로 가중치 값이 사라집니다. 이 상태에서 a = arr[x][2]를 하려고 할 때 런타임 에러가 납니다.