글 읽기 - Python3 시간 초과 질문입니다

comet15 3년 전 0

봄마다 나무 리스트를 sort 하는 방법과 애초에 나무 리스트를 heapq로 쓰는 방법 다 사용해 봤는데 둘다 시간초과가 나네요

어디서 시간을 줄일 수 있을까요?

import sys


class Area:
    def __init__(self, r, c, parent):
        self.r = r
        self.c = c
        self.trees = []
        self.nutur = 5
        self.parent = parent
        self.food = 0
        self.spread = 0

    def add_nurtur(self, nutur):
        self.nutur += nutur

    def add_trees(self, num):
        for i in range(num):
            self.trees.append(1)
            self.parent.trees += 1

    def spring_summer_winter(self):
        self.trees.sort(reverse=True)
        new_tree_list = []
        while self.trees:
            tree = self.trees.pop()
            if self.nutur >= tree:
                self.nutur -= tree
                tree += 1
                new_tree_list.append(tree)
                if tree % 5 == 0:
                    self.spread += 1
            else:
                self.nutur += int(tree / 2) + self.food
                self.parent.trees -= 1
        self.trees = new_tree_list
        self.nutur += self.food


class Yard:
    def __init__(self, n):
        self.land = [[Area(r, c, self) for c in range(n)] for r in range(n)]
        self.N = n
        self.trees = 0

    def set_food(self, r, c, f):
        self.land[r][c].food = f

    def spring_summer_winter(self):
        for line in self.land:
            for area in line:
                area.spring_summer_winter()

    def fall(self):
        for line in self.land:
            for area in line:
                if area.r > 0:
                    if area.c > 0:
                        self.land[area.r - 1][area.c - 1].add_trees(area.spread)
                    self.land[area.r - 1][area.c].add_trees(area.spread)
                    if area.c < self.N - 1:
                        self.land[area.r - 1][area.c + 1].add_trees(area.spread)
                if area.c > 0:
                    self.land[area.r][area.c - 1].add_trees(area.spread)
                    if area.r < self.N - 1:
                        self.land[area.r + 1][area.c - 1].add_trees(area.spread)
                if area.c < self.N - 1:
                    self.land[area.r][area.c + 1].add_trees(area.spread)
                    if area.r < self.N - 1:
                        self.land[area.r + 1][area.c + 1].add_trees(area.spread)
                if area.r < self.N - 1:
                    self.land[area.r + 1][area.c].add_trees(area.spread)
                area.spread = 0


N, M, K = map(int, sys.stdin.readline().split())
yard = Yard(N)
for n1 in range(N):
    foods = sys.stdin.readline().split()
    for n2 in range(N):
        yard.set_food(n1, n2, int(foods[n2]))
for m in range(M):
    x, y, z = map(int, sys.stdin.readline().split())
    yard.land[x - 1][y - 1].trees.append(z)
    yard.trees += 1

for k in range(K):
    yard.spring_summer_winter()
    yard.fall()

print(yard.trees)

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

문제

문제

출처

ICPC

Python3 시간 초과 질문입니다

comet15 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말