Dot Algo∙ DS/PS
[BOJ] 백준 14939번 불 끄기 (Java)
루지
2022. 3. 1. 14:05
#14939 불 끄기
난이도 : 플레 5
유형 : 브루트 포스 / 그리디 알고리즘
▸ 문제
전구 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];
}
}
}
}