글 읽기 - 시간초과가 나왔는데 해결 방법을 잘 모르겠습니다

donny0417 1년 전 0

함수 btime에 for문이 2중으로 들어가는데 그 함수를 for문 안에서 쓰다보니 시간이 오래걸리게 되는 것 같습니다

하지만 제 나름으로는 for문을 최대한으로 줄였다고 생각해서 혹시 for문을 줄일 수 있는 추가적인 방법이나

시간적으로 더 단축시킬 수 있는 방법이 있을지 궁금합니다

import sys

n,m,b = map(int, sys.stdin.readline().split())

blocks = []

for i in range(n):
    blocks.append(list(map(int, sys.stdin.readline().split())))

total = 0

for i in range(n):
    for j in range(m):
        total += blocks[i][j]
        
req = (m*n) - total % (m*n)

heights = range(((total + req)//(m*n)-1),(total + b)//(m*n))

def btime(x):
    time = 0
    for i in range(n):
        for j in range(m):
            if heights[x] - blocks[i][j] > 0:
                time += heights[x] - blocks[i][j]
            else:
                time += 2 * (blocks[i][j] - heights[x])
    return time

if req <= b:
    t,h = btime(0), heights[0]
    for i in range(len(heights)):
        if btime(i) <= t and heights[i] >= h:
            t,h = btime(i), heights[i]
    print(t,h,end=' ')
                
else:
    print(int(btime(0)),int(heights[0]),end=' ')

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

문제

문제

출처

ICPC

시간초과가 나왔는데 해결 방법을 잘 모르겠습니다

donny0417 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말