글 읽기 - 런타임 에러 (IndexError)

yongjin120 1년 전 0

jupyter notebook에서는 잘 돌아가는데 뭐가 문제일까요?

N, M, K = map(int, input().split(' '))
A = []
for _ in range(N):
    A.append(list(map(int, input().split())))

tree_pos = []
for _ in range(M):
    a,b,c = map(int, input().split())
    tree_pos.append([a-1,b-1,c])
tree_pos = sorted(tree_pos, key = lambda arr:arr[2], reverse= True)
tree_map = [[5]*N for _ in range(N)]

dx = [-1, -1, 0, 1, 1, 1, 0, -1]
dy = [0, 1, 1, 1, 0, -1, -1, -1]

for _ in range(K):
    dead_trees = []
    for _ in range(len(tree_pos)):    
        x, y, ag = tree_pos.pop()
        # 봄 
        if tree_map[x][y] < ag:
            dead_trees.append([x,y,ag])
        else:
            tree_map[x][y] -= ag
            tree_pos.insert(0,[x, y, ag+1])     
            
    # 여름
    while dead_trees:
        x,y,ag = dead_trees.pop()
        tree_map[x][y] += ag//2
    
    # 가을 # 겨울
    
    for i in range(len(tree_pos)):     
        x, y, ag = tree_pos[i]
        if ag % 5 == 0:
            for j in range(8):
                nx, ny = x + dx[j], y + dy[j]
                
                if nx < 0 or nx >= N or ny < 0 or ny >= N:
                    continue

                tree_pos.append([nx + 1, ny + 1, 1])
        
    for i in range(N):
        for j in range(N):
            tree_map[i][j] += A[i][j]
    
    
print(len(tree_pos))  

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

문제

문제

출처

ICPC

런타임 에러 (IndexError)

yongjin120 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말