[Java] 3055. 탈출

문제

사악한 암흑의 군주 이민혁은 드디어 마법 구슬을 손에 넣었고, 그 능력을 실험해보기 위해 근처의 티떱숲에 홍수를 일으키려고 한다. 이 숲에는 고슴도치가 한 마리 살고 있다. 고슴도치는 제일 친한 친구인 비버의 굴로 가능한 빨리 도망가 홍수를 피하려고 한다.

티떱숲의 지도는 R행 C열로 이루어져 있다. 비어있는 곳은 '.'로 표시되어 있고, 물이 차있는 지역은 '*', 돌은 'X'로 표시되어 있다. 비버의 굴은 'D'로, 고슴도치의 위치는 'S'로 나타내어져 있다.

매 분마다 고슴도치는 현재 있는 칸과 인접한 네 칸 중 하나로 이동할 수 있다. (위, 아래, 오른쪽, 왼쪽) 물도 매 분마다 비어있는 칸으로 확장한다. 물이 있는 칸과 인접해있는 비어있는 칸(적어도 한 변을 공유)은 물이 차게 된다. 물과 고슴도치는 돌을 통과할 수 없다. 또, 고슴도치는 물로 차있는 구역으로 이동할 수 없고, 물도 비버의 소굴로 이동할 수 없다.

티떱숲의 지도가 주어졌을 때, 고슴도치가 안전하게 비버의 굴로 이동하기 위해 필요한 최소 시간을 구하는 프로그램을 작성하시오.

고슴도치는 물이 찰 예정인 칸으로 이동할 수 없다. 즉, 다음 시간에 물이 찰 예정인 칸으로 고슴도치는 이동할 수 없다. 이동할 수 있으면 고슴도치가 물에 빠지기 때문이다.

입력

첫째 줄에 50보다 작거나 같은 자연수 R과 C가 주어진다.

다음 R개 줄에는 티떱숲의 지도가 주어지며, 문제에서 설명한 문자만 주어진다. 'D'와 'S'는 하나씩만 주어진다.

출력

첫째 줄에 고슴도치가 비버의 굴로 이동할 수 있는 가장 빠른 시간을 출력한다. 만약, 안전하게 비버의 굴로 이동할 수 없다면, "KAKTUS"를 출력한다.

예제 입력 1 

3 3

D.*

...

.S.

예제 출력 1 

3

예제 입력 2 

3 3

D.*

...

..S

예제 출력 2 

KAKTUS

예제 입력 3 

3 6

D...*.

.X.X..

....S.

예제 출력 3 

6

예제 입력 4 

5 4

.D.*

....

..X.

S.*.

....

예제 출력 4 

4

 

 

📌 나의 풀이

 

import java.io.*;
import java.util.*;

class Pos{
    int x;
    int y;
    public Pos(int x, int y){
        this.x=x;
        this.y=y;
    }
}

class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        String nm=br.readLine();
        StringTokenizer st=new StringTokenizer(nm, " ");
        int n=Integer.parseInt(st.nextToken());
        int m=Integer.parseInt(st.nextToken());

        int[][] visit=new int[n][m];
        String[][] lst=new String[n][m];
        Deque<Pos> que=new ArrayDeque<>();
        int[] d=new int[2];     // 도착 지점 행, 열
        int[] dx={1, -1, 0, 0};
        int[] dy={0, 0, 1, -1};
        boolean flag=false;

        // 지도 입력
        for (int i=0; i<n; i++){
            String tmp=br.readLine();

            for(int j=0; j<m; j++){
                String s=tmp.substring(j, j+1);
                if(s.equals("S")){
                    que.addFirst(new Pos(i, j));     // 시작 추가
                }
                else if(s.equals("*")){
                    que.add(new Pos(i, j));     // 물 추가
                }
                else if(s.equals("D")){
                    d[0]=i;
                    d[1]=j;  // 도착 추가
                }
                lst[i][j]=s;
            }
        }


        while (!que.isEmpty()){
            if(flag){
               break;
            }
            Pos xy=que.pollFirst();

            for(int i=0; i<4; i++){
                int nx=xy.x+dx[i];
                int ny=xy.y+dy[i];
                if(0>nx || nx>=n || ny<0 || ny>=m) continue;
                if(lst[xy.x][xy.y].equals("*")){      // 물
                    if(lst[nx][ny].equals("S") || lst[nx][ny].equals(".")){
                        lst[nx][ny]="*";
                        que.add(new Pos(nx,ny));
                    }
                }else if(lst[xy.x][xy.y].equals("S")){     // 고슴도치 위치
                    if(lst[nx][ny].equals(".")){
                        lst[nx][ny]="S";
                        que.add(new Pos(nx, ny));
                        visit[nx][ny]=visit[xy.x][xy.y]+1;
                    }else if(lst[nx][ny].equals("D")) {
                        flag=true;
                        visit[nx][ny]=visit[xy.x][xy.y]+1;
                        break;
                    }
                }
            }
        }

        if(visit[d[0]][d[1]] != 0){
            System.out.println(visit[d[0]][d[1]]);
        }else{
            System.out.println("KAKTUS");
        }
    }
}

다른 사람이 파이썬으로 푼 풀이를 자바로 구현해보았다.

자바로 구현하니 코드가 엄청 길다🤣

첨에 BufferedReader 대신 Scanner를 사용했는데 NoSuchElement 에러가 떠서..

hasNext 메소드로 입력의 끝을 파악하려고 했었다.

잘 안돼서 이참에 성능이 더 좋은 BufferedReader를 썼다.

 

 

# 시간 초과
from collections import deque

R, C=map(int, input().split())
map_lst=[list(input()) for i in range(R)]
visited=[[0]*C for _ in range(R)]
wq=deque()

for i in range(R) :
    for j in range(C) :
        if "S" == map_lst[i][j] :
            column=j
            row=i
        elif "*" == map_lst[i][j] :
            wq.append([i, j])


def water():
    wqlen=len(wq)
    while wqlen :
        x, y=wq.popleft()
        for dx, dy in (1, 0), (-1, 0), (0, 1), (0, -1) :
            nx, ny=x+dx, y+dy
            if 0<=nx<R and 0<=ny<C :
                if map_lst[nx][ny] == "." :
                    map_lst[nx][ny]="*"
                    wq.append([nx, ny])
        wqlen-=1


def bfs():
    que=deque([[row, column, 0]])

    while que :
        lenque=len(que)

        while lenque :
            x, y, depth=que.popleft()
            visited[x][y]=1

            for dx, dy in (-1, 0), (1, 0), (0, 1), (0, -1) :
                nx, ny=x+dx, y+dy
                if 0<=nx<R and 0<=ny<C and not visited[nx][ny]:
                    if map_lst[nx][ny] == "." :
                        que.append([nx, ny, depth+1])
                    elif map_lst[nx][ny] == "D" :
                        print(depth+1)
                        return
            lenque-=1
        water()

    print("KAKTUS")

water()
bfs()

시간 초과..

고슴도치가 이동하는 부분은 내가 짰는데 물이 차는 것은 모르겠어서 다른 사람 풀이를 참고했다.

입력에서 *(물)이 여러 개 주어질 수 있는데 어디부터 차게 해야하나를 몰랐었다.

*의 좌표를 deque에 넣는다.

큐의 길이만큼 반복문을 실행하는 것은 매 분을 표현하는 것이다.

 

 

🍟 다른 사람 풀이

# https://chldkato.tistory.com/22

from collections import deque

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

def bfs(x, y) :
    q.append([x, y])
    c[x][y]=1
    while q:
        qlen=len(q)
        while qlen :
            x, y=q.popleft()
            for i in range(4) :
                nx=x+dx[i]
                ny=y+dy[i]
                if 0 <= nx < n and 0 <= ny < m :
                    if a[nx][ny] == '.' and c[nx][ny] == 0 :
                        c[nx][ny]=c[x][y]+1
                        q.append([nx, ny])
                    elif a[nx][ny] == 'D' :
                        print(c[x][y])
                        return
            qlen-=1
        water()

    print("KAKTUS")
    return

def water():
    qlen=len(wq)
    while qlen :
        x, y=wq.popleft()
        for i in range(4) :
            nx=x+dx[i]
            ny=y+dy[i]
            if 0 <= nx < n and 0 <= ny < m :
                if a[nx][ny] == '.' :
                    a[nx][ny]='*'
                    wq.append([nx, ny])
        qlen-=1

n, m=map(int, input().split())
a=[list(map(str, input())) for _ in range(n)]
c=[[0]*m for _ in range(n)]
q, wq=deque(), deque()

for i in range(n) :
    for j in range(m) :
        if a[i][j] == 'S' :
            x1, y1=i, j
            a[i][j]='.'
        elif a[i][j] == '*' :
            wq.append([i, j])

water()
bfs(x1, y1)

내가 참고했던 풀이

 

 

# https://sinsomi.tistory.com/entry/%EB%B0%B1%EC%A4%80-Python-3055%EB%B2%88-%ED%83%88%EC%B6%9C-%EC%B4%88%EC%BD%94%EB%8D%94

import sys
from collections import deque

r,c=map(int, input().split())
maps=[list(p for p in sys.stdin.readline().strip()) for _ in range(r)]
visit=[[0]*c for _ in range(r)]
dr=[1, 0, -1, 0]
dc=[0, -1, 0, 1]
queue=deque()

# 물, 고슴도치 위치 queue에 담기
# 굴의 위치는 target에 저장
for i in range(r) :
    for j in range(c) :
        if maps[i][j] == '*' :
            queue.append([i, j])
        elif maps[i][j] == 'S' :
            queue.appendleft([i, j])
        elif maps[i][j] == 'D' :
            target_r=i
            target_c=j

flag=False
# 물과 고슴도치 위치에서 BFS 탐색
while queue :
    # 굴에 도착하고 나면 while문 탈출
    if flag :
        break
    pr, pc=queue.popleft()
    for i in range(4) :
        nr, nc=pr+dr[i], pc+dc[i]
        if nr < 0 or nr >= r or nc < 0 or nc >= c :
            continue

        # 물
        if maps[pr][pc] == '*' :
            if maps[nr][nc] == '.' or maps[nr][nc] == 'S' :
                maps[nr][nc]='*'
                queue.append([nr, nc])

        # 고슴도치
        elif maps[pr][pc] == 'S' :
            if maps[nr][nc] == '.' :
                maps[nr][nc]='S'
                visit[nr][nc]=visit[pr][pc]+1
                queue.append([nr, nc])
            # 굴에 도착하면 탈출
            elif maps[nr][nc] == 'D' :
                flag=True
                visit[nr][nc]=visit[pr][pc]+1
                break

# 굴에 도착하지 못하면 visit[굴]이 0이므로
if visit[target_r][target_c] == 0 :
    print('KAKTUS')
else : print(visit[target_r][target_c])

queue에 고슴도치의 위치, 물의 위치를 담는다.

고슴도치의 위치가 가장 맨 앞에 있어야 한다.

고슴도치의 위치 'S'에도 물이 찰 수 있다.

visit 리스트는 시간을 담고있다.

 

 

 

문제 출처 👉 백준