[프로그래머스] 프렌즈4블록

[프로그래머스] 프렌즈4블록

• https://programmers.co.kr/learn/courses/30/lessons/17679

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

//높이 M, 폭 N
int M, N;

vector<vector<int>> intBoard;
vector<vector<int>> eraseBoard;

bool inRange(int r, int w) {
	if (r < 0 || r >= M) return false;
	if (w < 0 || w >= N) return false;
	return true;
}

//(r, w)(r, w+1)
//(r+1, w)(r+1, w+1)
// 2x2 정사각형의 좌표 intBoard에서 모두 같은지 확인
bool findSquare(int r, int w) {
	if (!inRange(r, w + 1) || !inRange(r + 1, w) || !inRange(r + 1, w + 1)) return false;
	
	int cur = intBoard[r][w];
	if ((cur == intBoard[r][w + 1]) && (cur == intBoard[r + 1][w]) && (cur == intBoard[r + 1][w + 1])) return true;
	else return false;
}


//(r, w)(r, w+1)
//(r+1, w)(r+1, w+1)
// 2x2 정사각형의 좌표 eraseBoard에 체크하며, 새롭게 체크된 칸이 몇 개인지 반환
int eraseSquare(int r, int w) {
	int cnt = 0;
	
	if (eraseBoard[r][w] == 0) cnt++;
	if (eraseBoard[r][w+1] == 0) cnt++;
	if (eraseBoard[r+1][w] == 0) cnt++;
	if (eraseBoard[r+1][w+1] == 0) cnt++;

	eraseBoard[r][w] = eraseBoard[r][w+1] = eraseBoard[r+1][w] = eraseBoard[r+1][w+1] = 1;

	return cnt;
}

//(r, w)이 삭제된 경우 호출 
void moveBoardDown(int r, int w) {
	for (int i = r; i > 0 ; --i) {
		intBoard[i][w] = intBoard[i - 1][w];
	}
	intBoard[0][w] = -1;
	return;
}

int solution(int m, int n, vector<string> board) {

	//전역변수 초기화
	M = m;
	N = n;
	for (int i = 0; i < M; ++i) {
		vector<int> boardBuff;
		vector<int> eraseBuff;
		for (int j = 0; j < N; ++j) {
			boardBuff.push_back(board[i][j]);
			eraseBuff.push_back(0);
		}
		intBoard.push_back(boardBuff);
		eraseBoard.push_back(eraseBuff);
	}


	int answer = 0;

	while (true) {
		int eraseCnt = 0;
		//intBoard 순회하며 2x2 칸 찾기
		for (int r = 0; r < M; ++r) {
			for (int w = 0; w < N; ++w) {
				// intBoard 빈칸인 경우 스킵
				if (intBoard[r][w] == -1) continue;

				// 2x2 칸이 모두 같은지 검사
				// 모두 같다면 eraseBoard에 지워지는 칸 체크 & 새롭게 체크된 칸이 몇 개 인지 계산
				if (findSquare(r, w)) eraseCnt += eraseSquare(r, w);
			}
		}

		//지워진 칸 없는 경우 게임 종료
		if (eraseCnt == 0) break;
		else answer += eraseCnt;

		//intBoard 칸 위치 조정
		for (int w = 0; w < N; ++w) {
			for (int r = 0; r < M; ++r) {
				if (eraseBoard[r][w] == 1) {
					moveBoardDown(r, w);
					eraseBoard[r][w] = 0;
				}
			}
		}
	}

	return answer;
}