[백준] 14500번 벽 부수고 이동하기

HL·2021년 1월 17일
0

백준

목록 보기
15/104
  • 출처 : https://www.acmicpc.net/problem/14500

  • 알고리즘 : BFS

  • 풀이

    1. 큐를 사용해 BFS 구현
    2. 거리를 저장하는 배열에 벽을 (부수고 도달한 거리, 부수지 않고 도달한 거리)를 모두 저장
    3. -1로 초기화 (도달 X)
    4. 마지막 좌표의 두 거리 중 작은 것 출력
  • 소감

    • 구글링 코드 참고
    • 모두 이해는 했는데 혼자서 생각해낼 자신은 없다..

코드

from collections import deque

def init():

    n, m = map(int, input('').split(' '))

    board = []
    for i in range(n):
        line = list(map(int, list(input(''))))
        board.append(line)

    return n, m, board


def out(i, j):

    if 0 <= i < n and 0 <= j < m:
        return False
    return True


def print_line(line_list):
    for line in line_list:
        print(line)


n, m, board = init()

dist = [[[-1, -1] for j in range(m)] for i in range(n)]
dist[0][0] = [1 , 1]

start = (0,0,0)

q = deque()
q.append(start)

di = [1,-1,0,0]
dj = [0,0,1,-1]

while len(q) > 0:

    curr_i, curr_j, curr_b = q.popleft()

    for i in range(4):

        new_i = curr_i + di[i]
        new_j = curr_j + dj[i]

        if out(new_i, new_j):
            continue
        
        # 다음 칸이 빈칸
        if board[new_i][new_j] == 0:
            # 방문한 적 X
            if dist[new_i][new_j][curr_b] == -1:
                # 추가
                dist[new_i][new_j][curr_b] = dist[curr_i][curr_j][curr_b] + 1
                q.append((new_i, new_j, curr_b))

        # 다음 칸이 벽
        if board[new_i][new_j] == 1:
            # 부순적 X
            if curr_b == 0:
                # 부수고 방문한 적 X
                if dist[new_i][new_j][1] == -1:
                    # 추가
                    dist[new_i][new_j][1] = dist[curr_i][curr_j][0] + 1
                    q.append((new_i, new_j, 1))

# 도착 불가
if max(dist[n-1][m-1]) == -1:
    print(-1)
# 하나라도 도착하면 큰 것
elif min(dist[n-1][m-1]) == -1:
    print(max(dist[n-1][m-1]))
# 둘 다 도착하면 작은 것
else:
    print(min(dist[n-1][m-1]))
profile
Swift, iOS 앱 개발을 공부하고 있습니다

0개의 댓글