글 읽기 - 파이썬 3830 type error의 원인이 무엇인가요?

nayounsang1 1년 전 0

예제를 그대로 적용하면 오류가 없는데

실험의 개수(M)이 다 돌지 않더라도

0 0이 나오면 종료 될 수 있나요?

while True:
    N,M = map(int,input().split())
    if N==0 and M==0:#종료조건
        break
    parent = [[i,0] for i in range(N+1)] #parent[i][0] : i의 부모,[i][1]: i 보다 parent[i][1]만큼 무거움
    def union(pa,pb):
        if pa < pb:
            parent[pa][0] = pb
        else:
            parent[pb][0] = pa
    def spread(x):
        stack = [x]
        node_stack = [] #탐색한 노드
        val_stack = [] #시행마다의 가중치의 합
        val = 0#가중치의 합을 저장할 변수
        while stack:
            p = stack.pop()
            if p == parent[p][0]:#종료조건
                break
            node_stack.append(p)#탐색한 노드를 넣음
            val += parent[p][1]#가중치를 더함
            val_stack.append(val)
            stack.append(parent[p][0])
        val_data = [val]*len(val_stack)
        for i in range(len(val_stack)-1):#총합에서 이전가중치까지의 합만큼 뺌
            val_data[i+1] -= val_stack[i] 
        for n,v in zip(node_stack,val_data):
            parent[n][0] = p
            parent[n][1] = v#부모와 가중치 갱신
        return p
    def adjust(x,v):
        while True:
            xp = parent[x]
            if xp == x:
                return False#갱신시킬 곳 이 없다면
            if parent[x][1] == 0:#정의가 안된 곳이 있다면 입력한 가중치로 갱신시킴
                parent[x][1] = v
                return True
            v -= parent[x][1]
            x = xp
        return False
    for _ in range(M):
        comm = list(input().split())
        if len(comm) == 4:
            a,b,w = map(int,comm[1:])
            if a > b:
                a,b,w = b,a,-w
            pa,pb = spread(a),spread(b)
            if pa == pb:#이미 계산한 것
                continue
            else:#가중치 갱신
                union(pa,pb)
                if not adjust(a,w):
                    adjust(b,-w)
        else:
            a,b = map(int,comm[1:])
            pa,pb = spread(a),spread(b)
            if pa == pb:#a와 b의 공통부모가 c라하면 parent[a][1] = c-a,parent[b][1] = c-b
                #따라서 b-a는 -(c-b)+(c-a)
                print(parent[a][1]-parent[b][1])
            else:#불가능
                print('UNKNOWN')
        #print(*parent)
'''
예제
2 2
! 1 2 1
? 1 2
2 2
! 1 2 1
? 2 1
4 7
! 1 2 100
? 2 3
! 2 3 100
? 2 3
? 1 3
! 4 3 150
? 4 1
0 0
'''
'''
1
-1
UNKNOWN
100
200
-50
'''

bamgoesn 1년 전 0

adjust의 함수 정의에서 x가 리스트가 아닌 것 같은데, xp=parent[x]는 리스트이지 않나요?

코드를 돌려보진 않았는데 각 변수의 타입을 추적해보면 그게 조금 미심쩍습니다.

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

문제

문제

출처

ICPC

파이썬 3830 type error의 원인이 무엇인가요?

nayounsang1 1년 전 0

bamgoesn 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말