글 읽기 - 마지막 케이스에서 85가 왜 출력되지 않는지 모르겟습니다

compa513 3년 전 0

시간초과나는 코드입니다만, 알고리즘 자체는 틀리지 않았는 것 같은데

마지막케이스에서만 제대로 답이 나오지 않는 이유를 도통 모르겠습니다. 어느 부분에서 실수한 것일까요?

import sys
import copy
import math
input = sys.stdin.readline

dir = [[-1, 0],[-1, 1],[0, 1],[1, 1],[1, 0],[1, -1],[0, -1],[-1, -1]]
trees = [] # 나무 목록
eng = [] # 겨울마다 보충하는 자양분

N, M, K = map(int, input().split())
tree_map = [[5]*(N+1) for _ in range(N+1)] # 상도 땅의 자양분 잔량

def spring():
    global trees
    global tree_map
    for i in range(len(trees)):
        x = trees[i]["x"]
        y = trees[i]["y"]
        age = trees[i]["age"]
        if tree_map[x][y] >= age:
            tree_map[x][y] -= age
            trees[i]['age'] += 1
        else:
            trees[i]['live'] = False
            tree_map[x][y] += math.floor(trees[i]['age'] / 2)
    # print(trees)

def summer():
    global trees
    live_trees = []
    for _ in range(len(trees)):
        if trees[_]['live']:
            live_trees.insert(0, trees[_])
    trees = live_trees
    # print(trees)

def fall():
    global trees
    temp = copy.deepcopy(trees)
    for i in range(len(trees)):
        if trees[i]['age'] % 5 == 0: # 나이가 5배수고
            x = trees[i]['x']
            y = trees[i]['y']
            for j in range(len(dir)):
                dir_x = x + dir[j][0]
                dir_y = y + dir[j][1]
                if 1 <= dir_x <= N and 1 <= dir_y <= N: # 맵을 벗어나지 않는 한에서
                    temp.insert(0, {'x': dir_x, 'y': dir_y, 'age': 1, 'live': True})
    trees = temp


def winter():
    for i in range(N):
        for j in range(N):
            tree_map[i][j] += eng[i][j]

for _ in range(N):
    eng.append(list(map(int, input().split())))

for _ in range(M):
    x, y, age = map(int, input().split())
    trees.append({"x": x, "y": y, "age": age, "live": True})

for _ in range(K):
    spring()
    summer()
    fall()
    winter()

    # mmap = [[0]*(N+1) for x in range(N+1)]
    # for i in range(len(trees)):
    #     x = trees[i]['x']
    #     y = trees[i]['y']
    #     mmap[x][y] += 1
    # for i in range(N+1):
    #     print(mmap[i])
    # print('\n')


print(len(trees))

# print(trees)
# print(tree_map)

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

문제

문제

출처

ICPC

마지막 케이스에서 85가 왜 출력되지 않는지 모르겟습니다

compa513 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말