[백준]#2931 가스관

SeungBird·2021년 3월 26일
0

⌨알고리즘(백준)

목록 보기
301/401
post-custom-banner

문제

러시아 가스를 크로아티아로 운반하기 위해 자그레브와 모스코바는 파이프라인을 디자인하고 있다. 두 사람은 실제 디자인을 하기 전에 파이프 매니아 게임을 이용해서 설계를 해보려고 한다.

이 게임에서 유럽은 R행 C열로 나누어져 있다. 각 칸은 비어있거나, 아래 그림과 같은 일곱가지 기본 블록으로 이루어져 있다.

가스는 모스크바에서 자그레브로 흐른다. 가스는 블록을 통해 양방향으로 흐를 수 있다. '+'는 특별한 블록으로, 아래 예시처럼 두 방향 (수직, 수평)으로 흘러야 한다.

파이프 라인의 설계를 마친 후 두 사람은 잠시 저녁을 먹으러 갔다. 그 사이 해커가 침임해 블록 하나를 지웠다. 지운 블록은 빈 칸이 되어있다.

해커가 어떤 칸을 지웠고, 그 칸에는 원래 어떤 블록이 있었는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 유럽의 크기 R과 C가 주어진다. (1 ≤ R, C ≤ 25)

다음 R개 줄에는 C개 글자가 주어지며, 다음과 같은 글자로 이루어져 있다.

  • 빈칸을 나타내는 '.'
  • 블록을 나타내는 '|'(아스키 124), '-','+','1','2','3','4'
  • 모스크바의 위치를 나타내는 'M'과 자그레브를 나타내는 'Z'. 두 글자는 한 번만 주어진다.

항상 답이 존재하고, 가스의 흐름이 유일한 경우만 입력으로 주어진다, 또, 모스크바와 자그레브가 하나의 블록과 인접해 있는 입력만 주어진다. 또, 불필요한 블록이 존재하지 않는다. 즉, 없어진 블록을 추가하면, 모든 블록에 가스가 흐르게 된다.

출력

지워진 블록의 행과 열 위치를 출력하고, 어떤 블록이었는지를 출력한다.

예제 입력 1

3 7
.......
.M-.-Z.
.......

예제 출력 1

2 4 -

예제 입력 2

3 5
..1-M
1-+..
Z.23.

예제 출력 2

2 4 4

예제 입력 3

6 10
Z.1----4..
|.|....|..
|..14..M..
2-+++4....
..2323....
..........

예제 출력 3

3 3 |

풀이

이 문제는 단순 구현 문제로 고려해야할 조건이 많아서 코드가 길긴 하지만 크게 어렵지 않은 문제였다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;

public class Main {
    static int N;
    static int M;
    static char[][] map;
    static int[] dx = {-1, 1, 0, 0};
    static int[] dy = {0, 0, -1, 1};

    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");
        N = Integer.parseInt(input[0]);
        M = Integer.parseInt(input[1]);
        map = new char[N][M];
        Pair start = null;

        for(int i=0; i<N; i++) {
            String str = br.readLine();
            for(int j=0; j<M; j++) {
                map[i][j] = str.charAt(j);

                if(map[i][j]=='M') {
                    start = new Pair(i, j, -1);
                    map[i][j]='.';
                }

                if(map[i][j]=='Z')
                    map[i][j]='.';
            }
        }

        for(int i=0; i<4; i++) {
            int nx = start.x+dx[i];
            int ny = start.y+dy[i];

            if(nx<0 || nx>=N || ny<0 || ny>=M) continue;

            if(map[nx][ny]!='.') {
                bfs(start.x, start.y, i);
                break;
            }
        }
    }

    public static void bfs(int x, int y, int idx) {
        Queue<Pair> queue = new LinkedList<>();
        queue.add(new Pair(x, y, idx));

        loop:while(!queue.isEmpty()) {
            Pair temp = queue.poll();
            int nx = temp.x+dx[temp.idx];
            int ny = temp.y+dy[temp.idx];

            switch(map[nx][ny]) {
                case '|':
                    queue.add(new Pair(nx, ny, temp.idx));

                break;

                case '-':
                    queue.add(new Pair(nx, ny, temp.idx));

                break;

                case '+':
                    queue.add(new Pair(nx, ny, temp.idx));

                break;

                case '1': {
                    if(temp.idx==2) {
                        queue.add(new Pair(nx, ny, 1));
                    }

                    else {
                        queue.add(new Pair(nx, ny, 3));
                    }
                }
                break;

                case '2': {
                    if(temp.idx==1) {
                        queue.add(new Pair(nx, ny, 3));
                    }

                    else {
                        queue.add(new Pair(nx, ny, 0));
                    }
                }
                break;

                case '3': {
                    if(temp.idx==1) {
                        queue.add(new Pair(nx, ny, 2));
                    }

                    else {
                        queue.add(new Pair(nx, ny, 0));
                    }
                }
                break;

                case '4': {
                    if(temp.idx==3) {
                        queue.add(new Pair(nx, ny, 1));
                    }

                    else {
                        queue.add(new Pair(nx, ny, 2));
                    }
                }
                break;

                case '.': {
                    if(temp.idx==0) {
                        char ans = ' ';
                        int cnt = 0;

                        if(nx>=1 && map[nx-1][ny]!='.' && map[nx-1][ny]!='-' && map[nx-1][ny]!='2' && map[nx-1][ny]!='3') {
                            ans = '|';
                            cnt++;
                        }

                        if(ny>=1 && map[nx][ny-1]!='.' && map[nx][ny-1]!='|' && map[nx][ny-1]!='3' && map[nx][ny-1]!='4') {
                            ans = '4';
                            cnt++;
                        }

                        if(ny<M-1 && map[nx][ny+1]!='.' && map[nx][ny+1]!='|' && map[nx][ny+1]!='1' && map[nx][ny+1]!='2') {
                            ans = '1';
                            cnt++;
                        }

                        if(cnt==3)
                            System.out.println((nx+1)+" "+(ny+1)+" +");
                        else
                            System.out.println((nx+1)+" "+(ny+1)+" "+ans);

                        break loop;
                    }

                    else if(temp.idx==1) {
                        char ans = ' ';
                        int cnt = 0;

                        if(nx<N-1 && map[nx+1][ny]!='.' && map[nx+1][ny]!='-' && map[nx+1][ny]!='1' && map[nx+1][ny]!='4') {
                            ans = '|';
                            cnt++;
                        }

                        if(ny>=1 && map[nx][ny-1]!='.' && map[nx][ny-1]!='|' && map[nx][ny-1]!='3' && map[nx][ny-1]!='4') {
                            ans = '3';
                            cnt++;
                        }

                        if(ny<M-1 && map[nx][ny+1]!='.' && map[nx][ny+1]!='|' && map[nx][ny+1]!='1' && map[nx][ny+1]!='2') {
                            ans = '2';
                            cnt++;
                        }

                        if(cnt==3)
                            System.out.println((nx+1)+" "+(ny+1)+" +");
                        else
                            System.out.println((nx+1)+" "+(ny+1)+" "+ans);

                        break loop;
                    }

                    else if(temp.idx==2) {
                        char ans = ' ';
                        int cnt = 0;

                        if(ny>=1 && map[nx][ny-1]!='.' && map[nx][ny-1]!='|' && map[nx][ny-1]!='3' && map[nx][ny-1]!='4') {
                            ans = '-';
                            cnt++;
                        }

                        if(nx>=1 && map[nx-1][ny]!='.' && map[nx-1][ny]!='-' && map[nx-1][ny]!='2' && map[nx-1][ny]!='3') {
                            ans = '2';
                            cnt++;
                        }

                        if(nx<N-1 && map[nx+1][ny]!='.' && map[nx+1][ny]!='-' && map[nx+1][ny]!='1' && map[nx+1][ny]!='4') {
                            ans = '1';
                            cnt++;
                        }

                        if(cnt==3)
                            System.out.println((nx+1)+" "+(ny+1)+" +");
                        else
                            System.out.println((nx+1)+" "+(ny+1)+" "+ans);

                        break loop;
                    }

                    else {
                        char ans = ' ';
                        int cnt = 0;

                        if(ny<M-1 && map[nx][ny+1]!='.' && map[nx][ny+1]!='|' && map[nx][ny+1]!='1' && map[nx][ny+1]!='2') {
                            ans = '-';
                            cnt++;
                        }

                        if(nx>=1 && map[nx-1][ny]!='.' && map[nx-1][ny]!='-' && map[nx-1][ny]!='2' && map[nx-1][ny]!='3') {
                            ans = '3';
                            cnt++;
                        }

                        if(nx<N-1 && map[nx+1][ny]!='.' && map[nx+1][ny]!='-' && map[nx+1][ny]!='1' && map[nx+1][ny]!='4') {
                            ans = '4';
                            cnt++;
                        }

                        if(cnt==3)
                            System.out.println((nx+1)+" "+(ny+1)+" +");
                        else
                            System.out.println((nx+1)+" "+(ny+1)+" "+ans);

                        break loop;
                    }
                }
            }
        }
    }

    public static class Pair {
        int x;
        int y;
        int idx;

        public Pair(int x, int y, int idx) {
            this.x = x;
            this.y = y;
            this.idx = idx;
        }
    }
}
profile
👶🏻Jr. Programmer
post-custom-banner

0개의 댓글