[JAVA] BOJ 백준 20061 모노미노도미노 2

https://www.acmicpc.net/problem/20061

20061번: 모노미노도미노 2

전체 코드

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

public class BOJ_20061 {
	static int[][] map;
	static int result;
	public static void main(String[] args) throws Exception {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		int N = Integer.parseInt(bf.readLine());
		map = new int[10][10];
		for(int i=0; i<N; i++) {
			StringTokenizer st = new StringTokenizer(bf.readLine(), " ");
			int t = Integer.parseInt(st.nextToken());
			int x = Integer.parseInt(st.nextToken());
			int y = Integer.parseInt(st.nextToken());
			block(t, x, y);
		}
		int cnt = 0;
		for(int i=0; i<10; i++)
			for(int j=0 ;j<10; j++)
				if(map[i][j] == 1) cnt++;
		System.out.println(result);
		System.out.println(cnt);
	}
	static void block(int t, int x, int y) {
		if(t == 1) {
			// 파란색 영역
			int j = 0;
			for(j=4; j<=9; j++) {
				if(map[x][j] == 0) continue;
				break;
			}
			map[x][j-1] = 1;
			// 초록색 영역
			int i=0;
			for(i=4; i<=9; i++) {
				if(map[i][y] == 0) continue;
				break;
			}
			map[i-1][y] = 1;
		}
		else if(t == 2) {
			// 파란색 영역
			int j=0;
			for(j=5; j<= 9; j++) {
				if(map[x][j] == 0 && map[x][j-1] == 0) continue;
				break;
			}
			map[x][j-1] = 1; map[x][j-2] = 1;
			// 초록색 영역
			int i=0;
			for(i=4; i<= 9; i++) {
				if(map[i][y] == 0 && map[i][y+1] == 0) continue;
				break;
			}
			map[i-1][y] = 1; map[i-1][y+1] = 1;
		}
		else if(t == 3) {
			// 파란색 영역
			int j=0;
			for(j=4; j<=9; j++) {
				if(map[x][j] == 0 && map[x+1][j] == 0)
					continue;
				break;
			}
			map[x][j-1] = 1; map[x+1][j-1] = 1;
			// 초록색 영역
			int i=0;
			for(i=5; i<= 9; i++) {
				if(map[i][y] == 0 && map[i-1][y] == 0) continue;
				break;
			}
			map[i-1][y] = 1; map[i-2][y] = 1;
		}
		score();
	}
	static void score() {
		// 파란색 영역에서 가득 찬 열 찾기
		f: for(int j=6; j<=9; j++) {
			for(int i=0; i<4; i++) {
				if(map[i][j] == 0) continue f;
			}
			result++;
			for(int k = j; k>=5; k--) {
				for(int i=0; i<4; i++) {
					map[i][k] = map[i][k-1];
				}
			}
			for(int i=0; i<4; i++)
				map[i][4] = 0;
		}
		// 초록색 영역에서 가득 찬 행 찾기
		f: for(int i=6; i<=9; i++) {
			for(int j=0; j<4; j++) {
				if(map[i][j] == 0) continue f;
			}
			result++;
			for(int k = i; k>=5; k--) {
				map[k] = map[k-1];
			}
			map[4] = new int[10];
		}
		// 연한색 영역에 칸 비우기
		// 1. 파란색
		f: for(int j=4; j<=5; j++) {
			boolean flag = false;
			for(int i=0; i<4; i++) {
				if(map[i][j] == 1) {flag = true; break;}
			}
			if(flag) {
				for(int k = 9; k>j; k--) {
					for(int i=0; i<4; i++) {
						map[i][k] = map[i][k-1];
					}
				}
				for(int i=0; i<4; i++)
					map[i][j] = 0;
				}
			}
		// 2. 초록색
		f: for(int i=4; i<=5; i++) {
			boolean flag = false;
			for(int j=0; j<4; j++) {
				if(map[i][j] == 1) {flag = true; break;}
			}
			if(flag) {
				for(int k = 9; k>i; k--) {
					map[k] = map[k-1];
				}
				map[i] = new int[10];
			}
		}
	}
}

block() 은 입력받은 블록을 파란색 영역과 초록색 영역에 놓아주는 역할을 한다.

score() 는 꽉 찬 행과 열을 확인한 뒤 제거하여 점수를 얻고 연한색 영역에 블록이 놓여있는지 확인한 뒤 미뤄주는 역할을 한다.

한 줄씩 보자!

static int[][] map;
static int result;

전역 변수처럼 사용할 변수들은 다음과 같다. 일단 map이라는 이차원 배열을 통해 블록이 놓여있는 부분은 1, 아닌 부분은 0으로 표시할 것이다.

result를 통해 점수를 저장할 것이다.

map = new int[10][10];
for(int i=0; i<N; i++) {
	StringTokenizer st = new StringTokenizer(bf.readLine(), " ");
	int t = Integer.parseInt(st.nextToken());
	int x = Integer.parseInt(st.nextToken());
	int y = Integer.parseInt(st.nextToken());
	block(t, x, y);
}

for문을 통해 block 정보를 받아서 함수를 실행한다. 이때 입력받은 값도 인덱스가 0부터 시작하므로 따로 처리해주지 않아도 된다.

int cnt = 0;
for(int i=0; i<10; i++)
	for(int j=0 ;j<10; j++)
		if(map[i][j] == 1) cnt++;
System.out.println(result);
System.out.println(cnt);

블록을 옮기고 점수 계산하는 것이 끝나게 되면 이중 for문으로 블록을 세준다. 그 뒤 점수와 남은 블록의 개수를 출력한다.

static void block(int t, int x, int y) {
		if(t == 1) {
			// 파란색 영역
			int j = 0;
			for(j=4; j<=9; j++) {
				if(map[x][j] == 0) continue;
				break;
			}
			map[x][j-1] = 1;
			// 초록색 영역
			int i=0;
			for(i=4; i<=9; i++) {
				if(map[i][y] == 0) continue;
				break;
			}
			map[i-1][y] = 1;
		}

블록이 1인 경우는 1x1짜리 블록이다. 우리는 파란색 영역에서 빨간색 영역이 가까운 곳부터 조사하여 어디에 블록을 놓아야할 지 결정할 것이다. 뒤에서 언급하겠지만 뒤에서부터 조사하게 되면 엉뚱한 곳에 블록을 놓게 됨을 유의하자!

for문 밖에서도 j의 값을 사용할 것이기 때문에 i, j는 for문 밖에서 선언한 뒤 사용해준다. 앞에서 부터 조사하며 처음 0이아닌 값이 나오는 곳을 만나면 for문을 멈춘다. 이때 블록이 점점 내려오다가 x, j칸을 만나고 블록은 이동을 멈추게 된다.

그러므로 블록의 위치는 그보다 한 칸 앞인 x, j-1칸이 된다.

초록색 영역도 마찬가지이므로 넘어간다.

else if(t == 2) {
	// 파란색 영역
	int j=0;
	for(j=5; j<= 9; j++) {
		if(map[x][j] == 0 && map[x][j-1] == 0) continue;
		break;
	}
	map[x][j-1] = 1; map[x][j-2] = 1;
	// 초록색 영역
	int i=0;
	for(i=4; i<= 9; i++) {
		if(map[i][y] == 0 && map[i][y+1] == 0) continue;
		break;
	}
	map[i-1][y] = 1; map[i-1][y+1] = 1;
}

마찬가지로 i, j는 for 문 밖에서 선언해준다.

파란색 영역은 아까와 달리 5부터 조사를 시작하는데 그 이유는 x, j칸과 x, j-1칸을 동시에 조사하기 때문이다. j를 4부터 시작하려면 조건식을 8로 변경하고 x, j칸과 x, j+1칸을 조사하면 된다. 위에서와 같이 두 칸이 0이면 블록을 놓을 수 있는 칸이므로 계속 continue를 한다. 처음으로 블록을 놓을 수 없는 칸을 만나면 블록은 더 이상 뒤로 갈 수 없게 된다. 따라서 그 전칸에 블록을 놓아준다.

초록색 영역도 마찬가지이다.

t가 2일 때와 3일 때는 크게 다르지 않기 때문에 3의 설명은 생략할 것이다.

static void score() {
		// 파란색 영역에서 가득 찬 열 찾기
		f: for(int j=6; j<=9; j++) {
			for(int i=0; i<4; i++) {
				if(map[i][j] == 0) continue f;
			}
			result++;
			for(int k = j; k>=5; k--) {
				for(int i=0; i<4; i++) {
					map[i][k] = map[i][k-1];
				}
			}
			for(int i=0; i<4; i++)
				map[i][4] = 0;
		}

이제 점수를 계산해줄 것이다. 제거 해줄 열은 4칸이 모두 1인 열이다.

if문을 통해서 4칸 중 하나라도 0인 값이 나오면 (블록이 없는 칸) continue를 통해 다음 열을 조사하게 한다.

이때 4칸이 모두 1이면 continue되지 않고 밑에 있는 코드를 실행하게 된다.

따라서 점수를 증가시켜주고 블록을 한칸씩 뒤로 밀어준다. 4열은 전부 밀려났기 때문에 0으로 채운다.

// 초록색 영역에서 가득 찬 행 찾기
f: for(int i=6; i<=9; i++) {
	for(int j=0; j<4; j++) {
		if(map[i][j] == 0) continue f;
	}
	result++;
	for(int k = i; k>=5; k--) {
		map[k] = map[k-1];
	}
	map[4] = new int[10];
}

파란색 영역과 초록색 영역은 크게 다른 점이 없다. 블록을 밀어주는 코드가 비교적 간단하다.

// 연한색 영역에 칸 비우기
// 1. 파란색
f: for(int j=4; j<=5; j++) {
	boolean flag = false;
	for(int i=0; i<4; i++) {
		if(map[i][j] == 1) {flag = true; break;}
	}
	if(flag) {
		for(int k = 9; k>j; k--) {
			for(int i=0; i<4; i++) {
				map[i][k] = map[i][k-1];
			}
		}
		for(int i=0; i<4; i++)
			map[i][j] = 0;
		}
	}

for문을 통해 4칸 중에 하나라도 1이라면 flag를 true로 바꾸고 for문을 멈춘다. 조사하는 열에 블록이 존재한다는 뜻이므로 아까와 같이 한칸씩 뒤로 밀어준다. 그 뒤 해당 열을 0으로 채운다.

위의 코드와 다른 점은 for문 밑에 있는 코드가 전부 0으로 채워져있을 때에도 실행되기 때문에 flag를 통해 구분점을 줘야한다는 것 뿐이다.

💡 TIP : 뒤에서부터 채우게 되면 2로 표현되야 하는 게 1로 표현된다.

1번

2번