백준 1516. 게임 개발 파이썬(PYTHON)

문제링크 : www.acmicpc.net/problem/1516

1516번: 게임 개발

 

def solution(lists):
    answer = [0]*len(lists)
    # 먼저 짓는게 없는 경우
    for i in range (len(lists)):
        if len(lists[i]) == 2:
            answer[i] = lists[i][0]
    while 0 in answer:
        for i in range(len(lists)):
            buildtime = lists[i][0]
            buildtimes=list()
            if answer[i] !=0:
                continue
            else:
                for j in range(1,len(lists[i])):
                    if j == len(lists[i])-1:
                        answer[i]=max(buildtimes)
                    else:
                        if answer[lists[i][j]-1]!=0:
                            buildtimes.append(buildtime+answer[lists[i][j]-1])
                        else:
                            break

    return answer


def main():
    N = int(input())
    lists = list()
    for i in range(N):
        build = list(map(int,input().split()))
        lists.append(build)
    
    answer = solution(lists)
    for i in answer:
        print(i)

main()