41-pknujsp

[pknujsp]

🔗 문제 링크

알고스팟

✔️ 소요된 시간

30분

✨ 수도 코드

벽을 최소한으로 부수면서 목적지까지 이동하는 경우를 찾는 문제

1. 중복 방문 방지

• visited 리스트를 사용해 각 위치의 방문 여부를 기록

2. 최소 힙을 사용해 상하좌우 탐색

• 현재까지 발견된 경로 중에서 가장 적은 비용(여기서는 벽을 가장 적게 부순 횟수)을 가진 경로를 먼저 처리한다
• 다익스트라 알고리즘

3. 탐색 : 목적지에 도착할 때 까지 탐색한다

• (벽 파괴 횟수, 위치) 를 원소로 하여 힙에 추가/삭제하며 탐색한다
• 목적지에 도착하면 벽 파괴 횟수를 출력하고 종료함

1. 주변 위치 확인하고, 이동하기 위해 힙에 추가
   • 현재 위치에서 상하좌우에 있는 위치를 확인한다
   • 이 때 다음 조건을 만족하는 위치에 대해서 힙에 추가하고 탐색한다
   • 미로의 범위 안에 있어야 하고
   • 아직 방문하지 않은 위치여야 한다
2. 힙에 추가할 때는, (새로운 비용, 새로운 위치) 로 원소를 추가한다
   1. 새로운 비용 : 현 위치까지의 벽 파괴 횟수 + 해당 위치가 벽이라면 1, 빈 방이라면 0
   2. 새로운 벽 파괴 횟수를 계산함

C, R = map(int, stdin.readline().split())
arr = [list(map(int, list(stdin.readline().strip()))) for _ in range(R)]

drc = ((1,0),(-1,0),(0,1),(0,-1))
visited = [[False] * C for _ in range(R)]
heap = [(0, 0, 0)]
target_r = R - 1
target_c = C - 1

while heap:
    cost, r, c = heappop(heap)

    if r == target_r and c == target_c:
        print(cost)
        break
    if visited[r][c]:
        continue      
        
    visited[r][c] = True
    
    for dr, dc in drc:
        nr = r + dr
        nc = c + dc

        if not 0 <= nr < R or not 0 <= nc < C:
            continue
        if visited[nr][nc]:
            continue
            
        heappush(heap, (cost + arr[nr][nc], nr, nc))

📚 새롭게 알게된 내용

[tgyuuAn]

import sys
from heapq import *

def input(): return sys.stdin.readline().rstrip()

N, M = map(int, input().split())
board = []
for _ in range(M): 
    board.append(list(input()))

heap = [(0, 0, 0)]

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

visited = set()
visited.add((0,0))

while heap:
    now_cost, now_x, now_y = heappop(heap)

    if (now_x, now_y) == (N-1, M-1): 
        print(now_cost)
        break

    for dir in range(4):
        new_x = now_x + dx[dir]
        new_y = now_y + dy[dir]

        if new_x < 0 or new_x >= N: continue
        if new_y < 0 or new_y >= M: continue
        if (new_x, new_y) in visited: continue
        
        if board[new_y][new_x] == "1": heappush(heap, (now_cost+1, new_x, new_y))
        else: heappush(heap, (now_cost, new_x, new_y))
        visited.add((new_x, new_y))

이 문제 보자마자 저번에 준성님이 추천해주신 일요일 아침의 데이트 문제 생각나서 바로 힙을 떠올릴 수 있었습니다!!

인사이트 감사합니다~~~~

[H0ngJu]

이번 PR로 다익스트라 다시 복습하고 갑니다 !

다익스트라를 구현한 다른 레퍼런스를 보니 확실히 heap으로 구현한게 코드도 짧고 시간복잡도도 줄어드네용 태규님 벨만-포드 pr과 함께 최단거리 알고리즘 공부하고 갑니닷 !!

준성님 소마 합격 축하드립니다 ~~ 🎉🎉

[H0ngJu]

준성님 코드 볼 때마다 느끼는건데 continue문을 잘 활용하시는 것 같습니다

만약에 visited[ ][ ]가 False인 경우에만 22~31번줄을 실행하도록 하면
저는 if-else로 구현했을 것 같은데 continue를 사용하니까 복잡한 인덴트까지 생각 안해도 되니 제 눈에는 이게 더 깔끔해 보이는 것 같기도 하네요 ! 😎