글 읽기 - 시간초과 어떻게 해결하면 좋을까요

jaehyeoklim 1년 전 0

음..어떤 부분을 고치거나 추가해야할지 조언 부탁드립니다.,,

import sys
from collections import deque

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

def run():
    while rq:
        x, y = rq.popleft()
        for i in range(4):
            nx, ny = x + dx[i], y + dy[i]
            nk = 1
            while nk <= K and 0 <= nx < N and 0 <= ny < M and gym[nx][ny] != '#' and dist[nx][ny] < dist[x][y]:
                if dist[nx][ny] == -1:
                    rq.append([nx, ny])
                    dist[nx][ny] = dist[x][y] + 1
                nx += dx[i]; ny += dy[i]; nk += 1 
    return dist[x2][y2]

if __name__ == "__main__":
    N, M, K = map(int, sys.stdin.readline().split())
    gym = [list(map(str, sys.stdin.readline().rstrip())) for _ in range(N)]
    x1, y1, x2, y2 = map(int, sys.stdin.readline().split())
    x1 -= 1; y1 -= 1; x2 -= 1; y2 -= 1
    dist = [[-1] * M for _ in range(N)]
    dist[x1][y1] = 0

    rq = deque()
    rq.append([x1, y1])

    print(run())

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

문제

문제

출처

ICPC

시간초과 어떻게 해결하면 좋을까요

jaehyeoklim 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말