본문 바로가기

Dot Algo∙ DS/PS

[BOJ] 백준 14939번 불 끄기 (Java)

#14939 불 끄기

난이도 : 플레 5

유형 : 브루트 포스 / 그리디 알고리즘

 

14939번: 불 끄기

전구 100개가 10×10 정사각형 모양으로 늘어서 있다. 전구에 달린 스위치를 누르면 그 전구와 위, 아래, 왼쪽, 오른쪽에 있는 전구의 상태도 바뀐다. 전구 100개의 상태가 주어지면 모든 전구를 끄

www.acmicpc.net

▸ 문제

전구 100개가 10×10 정사각형 모양으로 늘어서 있다. 전구에 달린 스위치를 누르면 그 전구와 위, 아래, 왼쪽, 오른쪽에 있는 전구의 상태도 바뀐다. 전구 100개의 상태가 주어지면 모든 전구를 끄기 위해 최소한으로 눌러야 하는 스위치의 개수를 출력하라

 입력

10줄에 10글자씩 입력이 주어진다. #은 꺼진 전구고 O(대문자 알파벳 o)는 켜진 전구다. #과 O외에는 입력으로 주어지지 않는다.

 출력

모든 전구를 끄기 위해 최소한으로 눌러야 하는 스위치의 개수를 출력하라. 불가능하면 -1를 출력하라.

 

문제 풀이  

최소한의 스위치를 눌러서 모든 전구의 수를 끌 수 있는 경우를 구해야 한다. 모든 전구를 끄기 위해서는 첫 번째 열부터 모든 경우의 수를 고려하여(1<<10) 하나씩 꺼나가면 된다.

  • 첫번째 열(y=1)을 전구를 모든 경우의 수(1<<10)로 눌러본다.
  • y=2열 부터 이전 열(y-1)에 전구가 커져있는 경우만 선택한다. 그러면 1~y-1열까지의 모든 전구는 꺼져있다고 볼 수 있다. 그렇게 y=10열까지 반복한다.
  • 모든 열을 탐색한 후 마지막 10열에 커져있는 전구가 있다면 불가능한 경우로 간주하고 만약 10열의 전구가 모두 꺼져있다면 1~9열의 전구 또한 꺼져있으므로 cnt를 기록한다.

 

풀이 코드 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

	static final int SIZE = 10;
	static final int INF = Integer.MAX_VALUE;
	static int[] dx = {0, 0, -1, 1};
	static int[] dy = {-1, 1, 0, 0};
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		char[][] map = new char[SIZE][SIZE];
		for(int i=0; i<SIZE; i++) {
			map[i] = br.readLine().toCharArray();
		}
		
		int answer = INF;
		char[][] tmp = new char[SIZE][SIZE];
		
		// 0~9 버튼 누를 모든 경우의 수 
		for(int c=0; c<(1<<10); c++) {
			deepCopy(map, tmp);
			
			int cnt = 0;
			cnt = simulation(tmp, c, cnt);
			
			if(isAllOff(tmp)) {
				answer = Math.min(answer, cnt);
			}
		}
		
		System.out.println(answer == INF ? -1 : answer);
	}

	private static boolean isAllOff(char[][] tmp) {
		boolean flag = true;
		for(int x=0; x<10; x++) {
			if(tmp[9][x] == 'O') {
				flag = false;
				break;
			}
		}
		return flag;
	}

	private static int simulation(char[][] tmp, int c, int cnt) {
		for(int x=0; x<10; x++) {
			if((c & (1<<x)) != 0) {
				lightSwitch(tmp, x, 0);
				cnt++;
			}
		}
		
		for(int y=1; y<10; y++) {
			for(int x=0; x<10; x++) {
				if(tmp[y-1][x] == 'O') {
					lightSwitch(tmp, x, y);
					cnt++;
				}
			}
		}
		return cnt;
	}
	
	static void lightSwitch(char[][] map, int x, int y) {
		map[y][x] = switchOnOff(map[y][x]);
		for(int d=0; d<4; d++) {
			int nx = x + dx[d];
			int ny = y + dy[d];
			
			if(nx <0 || ny <0 || nx >= SIZE || ny >= SIZE) continue;
			
			map[ny][nx] = switchOnOff(map[ny][nx]);
		}
	}
	
	static char switchOnOff(char status) {
		if(status == '#') return 'O';
		else return '#';
	}
	
	static void deepCopy(char[][] src, char[][] copy) {
		for(int i=0; i<SIZE; i++) {
			for(int j=0; j<SIZE; j++) {
				copy[i][j] = src[i][j];
			}
		}
	}
}