[JAVA][BOJ][G3] 2931. 가스관

2931번: 가스관

문제

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

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

블록 '|'	블록 '-'	블록 '+'	블록 '1'	블록 '2'	블록 '3'	블록 '4'

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

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

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

입력

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

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

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

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

출력

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

풀이

함수 덜 쓰는 버릇 고쳐야하는데 마음처럼 안된다... 조건만 잘 세우면 사방탐색과 if문을 잘 섞어서 풀 수 있다 조건들이 많이 때문에 코드에 주석으로 설명을 달아뒀다

코드

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

public class g3_2931_가스관 {

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		int R = Integer.parseInt(st.nextToken());
		int C = Integer.parseInt(st.nextToken());
		
		char[][] arr = new char[R][C];
		
		for (int i = 0; i < R; i++) {
			String str = br.readLine();
			for (int j = 0; j < C; j++) {
				arr[i][j] = str.charAt(j);
			}
		}
		
		// 북 남 서 동
		int[] dx = {-1, 1, 0, 0};
		int[] dy = {0, 0, -1, 1};
		
		// 파이프의 모양
		char[] pipe = new char[4];
		// 파이프 위치를 찾았는지
		boolean find = false;
		
		// M과 Z가 제대로 연결되어 있는지 찾기
		boolean pipeM = false;
		boolean pipeZ = false;

		for (int i = 0; i < R; i++) {
			for (int j = 0; j < C; j++) {
				if(arr[i][j]=='M') {
					for (int k = 0; k < 4; k++) {
						int x = i+dx[k];
						int y = j+dy[k];

						if(x<0 || x>=R || y<0 || y>=C || arr[x][y]=='.') continue;
						
						pipeM = true;
					}
				} else if(arr[i][j]=='Z') {
					for (int k = 0; k < 4; k++) {
						int x = i+dx[k];
						int y = j+dy[k];

						if(x<0 || x>=R || y<0 || y>=C || arr[x][y]=='.') continue;
						
						pipeZ = true;
					}
				}
			}
		}
		
		// 지워진 블록 찾기
		findPipe: for (int i = 0; i < R; i++) {
			for (int j = 0; j < C; j++) {
				if(arr[i][j]=='.') {
					// 만약 연결되어 있지 않다면 새로 연결해야 함
					boolean connectM = pipeM;
					boolean connectZ = pipeZ ;
					
					// 사방탐색
					for (int k = 0; k < 4; k++) {
						int x = i+dx[k];
						int y = j+dy[k];

						if(x<0 || x>=R || y<0 || y>=C || arr[x][y]=='.') continue;
						
						if(arr[x][y] == '+') {
							pipe[k] = 1;
							find = true;
						}
						
						// M이나 Z가 연결되어 있지 않고, 맞닿아 있다면 연결
						// 만약 다른 파이프와 연결되지 않았다면 M이나 Z와도 연결되면 안 됨
						if((!pipeM && arr[x][y] == 'M') || (!pipeZ && arr[x][y] =='Z')) {
							pipe[k] = 1;
							if(arr[x][y] == 'M') connectM = true;
							else if(arr[x][y] == 'Z') connectZ = true;
						}
						
						switch(k) {
							case 0:
								if(arr[x][y] == '|' || arr[x][y] == '1' || arr[x][y] == '4') {
									pipe[0] = 1;
									find = true;
								}
								break;
							case 1:
								if(arr[x][y] == '|' || arr[x][y] == '2' || arr[x][y] == '3') {
									pipe[1] = 1;
									find = true;
								}
								break;
							case 2:
								if(arr[x][y] == '-' || arr[x][y] == '1' || arr[x][y] == '2') {
									pipe[2] = 1;
									find = true;
								}
								break;
								
							case 3:
								if(arr[x][y] == '-' || arr[x][y] == '3' || arr[x][y] == '4') {
									pipe[3] = 1;
									find = true;
								}
								break;
						}
					}
					
					// M과 Z가 연결되어 있고 그 외의 파이프와 연결된 상태라면 블록 위치 출력
					if(connectM && connectZ && find) {
						sb.append(i+1).append(" ").append(j+1).append(" ");
						break findPipe;
					}
					
					// 찾지 못했다면 다시 처음부터
					find = false;
					pipe = new char[4];
					
				}
			}
		}
		
		char[][] block = {{'|', 1, 1, 0, 0}, {'-', 0, 0, 1, 1}, {'+', 1, 1, 1, 1},
							{'1', 0, 1, 0, 1}, {'2', 1, 0, 0, 1}, {'3', 1, 0, 1, 0}, {'4', 0, 1, 1, 0}};

		// 파이프 모양 찾기
		for(int i = 0; i < 7; i++) {
			char[] tmp = new char[4];
			for(int j = 1; j < 5; j++) {
				tmp[j-1] = block[i][j];
			}
			if(Arrays.equals(pipe, tmp)) {
				sb.append(block[i][0]);
				break;
			}
		}
		
		System.out.println(sb);

	}

}

결과

 

#자바 #java #boj #백준 #알고리즘