글 읽기 - 어디가 틀린 걸까요?...

rbcjswkd 4년 전 0

웬만한 건 다 나오는 것 같은데 어디가 틀린지 모르겠네요;;ㅠ

from collections import deque
m_dir = [(1,0),(0,1),(-1,0),(0,-1)]
h_dir = [(2,1),(1,2),(2,-1),(1,-2),(-2,1),(-1,2),(-2,-1),(-1,-2)]

def BFS():
    while Q:
        r,c,h_cnt,cnt = Q.popleft()
        for dr,dc in m_dir:
            nr = r+dr
            nc = c+dc
            if nr>H-1 or nc>W-1 or nr<0 or nc<0 or arr[nr][nc]:continue
            if visit[nr][nc][0]:
                if visit[nr][nc][1]<=h_cnt:continue
            if (nr, nc) == (H - 1, W - 1):
                return cnt + 1
            Q.append((nr,nc,h_cnt,cnt+1))
            visit[nr][nc][0] = cnt+1
            visit[nr][nc][1] = h_cnt
        if h_cnt<K:
            for dr, dc in h_dir:
                nr = r+dr
                nc = c+dc
                if nr>H-1 or nc>W-1 or nr<0 or nc<0 or arr[nr][nc]:continue
                if visit[nr][nc][1] and visit[nr][nc][1] <= h_cnt: continue
                if visit[nr][nc][0]:
                    if visit[nr][nc][0]<=cnt:continue
                if (nr,nc) == (H-1,W-1):
                    return cnt+1
                Q.append((nr,nc,h_cnt+1,cnt+1))
                visit[nr][nc][0] = cnt+1
                visit[nr][nc][1] = h_cnt+1
    return -1

K = int(input())
W, H = map(int, input().split())
arr = [list(map(int,input().split())) for _ in range(H)]
visit = [[[0,0] for _ in range(W)] for _ in range(H)]
visit[0][0][0] = 1
Q = deque()
Q.append((0,0,0,0))
print(BFS())

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

문제

문제

출처

ICPC

어디가 틀린 걸까요?...

rbcjswkd 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말