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

프렌즈4블록

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 프렌즈4블록.
같은 모양의 카카오프렌즈 블록이 2×2 형태로 4개가 붙어있을 경우 사라지면서 점수를 얻는 게임이다.

만약 판이 위와 같이 주어질 경우, 라이언이 2×2로 배치된 7개 블록과 콘이 2×2로 배치된 4개 블록이 지워진다. 같은 블록은 여러 2×2에 포함될 수 있으며, 지워지는 조건에 만족하는 2×2 모양이 여러 개 있다면 한꺼번에 지워진다.

블록이 지워진 후에 위에 있는 블록이 아래로 떨어져 빈 공간을 채우게 된다.

만약 빈 공간을 채운 후에 다시 2×2 형태로 같은 모양의 블록이 모이면 다시 지워지고 떨어지고를 반복하게 된다.

위 초기 배치를 문자로 표시하면 아래와 같다.

TTTANT RRFACC RRRFCC TRRRAA TTMMMF TMMTTJ

각 문자는 라이언(R), 무지(M), 어피치(A), 프로도(F), 네오(N), 튜브(T), 제이지(J), 콘(C)을 의미한다

입력으로 블록의 첫 배치가 주어졌을 때, 지워지는 블록은 모두 몇 개인지 판단하는 프로그램을 제작하라.

입력 형식

• 입력으로 판의 높이 m, 폭 n과 판의 배치 정보 board가 들어온다.
• 2 ≦ n, m ≦ 30
• board는 길이 n인 문자열 m개의 배열로 주어진다. 블록을 나타내는 문자는 대문자 A에서 Z가 사용된다.

출력 형식

입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.

 

 

풀이 :

구현 문제.

배열에서 밑으로 위에 담긴 값들을 한 방향으로 미는 비슷한 문제들이, SWEA(모의 역량 테스트 : 벽돌깨기)나 백준(2048...)에 있다. 

처음에, 예제를 안보고, 문제만 읽어, '각 문자는 라이언(R), 무지(M), 어피치(A), 프로도(F), 네오(N), 튜브(T), 제이지(J), 콘(C)을 의미한다'에 따라 값을 배열에 넣을 때, 매칭해서 시간을 날렸다. 

1. 모든 점에서 자신을 포함해 2x2칸이 같은 것을 만족하면, set에 좌표를 넣어준다.

2. set에 있는 좌표를 배열에서 모두 .으로 바꿔준다.

3. 아래로 블록을 당긴다.

4. 1,2,3을 반복하며, set에 아무 좌표도 들어가지 않으면(만족하는 2x2가 없음) 위의 루프를 종료한다.

각 회당 set에 들어가 있던 좌표들의 수가 삭제되는 블록의 수이다.

#include <string>
#include <vector>
#include <set>
using namespace std;
char arr[31][31];
int dx[3] = {0, 1, 1};
int dy[3] = {1, 0, 1};
 
int solution(int m, int n, vector<string> board) {
    int answer = 0;
    
    for(int i = 0; i < m; i++){
        for(int j = 0; j < n; j++){
            arr[i][j] = board[i][j];
        }
    }
    
    while(1){
        set<pair<int, int>> s;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                char now = arr[i][j];
                if(now == '.') continue;
                int cnt = 0;
                for(int d = 0; d < 3; d++){
                    int nx = i + dx[d];
                    int ny = j + dy[d];
                    if(nx >= 0 && nx < m && ny >= 0 && ny < n){
                        if(now == arr[nx][ny]) cnt++;
                        else break;
                    }
                }
                if(cnt == 3){
                    s.insert({i, j});
                    for(int d = 0; d < 3; d++){
                        s.insert({i + dx[d], j + dy[d]});
                    }
                }
            }
        }
        
        if(s.size() == 0) break;
        for(pair<int,int> p : s){
            arr[p.first][p.second] = '.';
            answer++;
        }
        
        for(int i = 0; i < n; i++){
            for(int j = m-1; j >= 1; j--){
                if(arr[j][i] == '.'){
                    for(int k = j-1; k >= 0; k--){
                        if(arr[k][i] != '.'){
                            arr[j][i] = arr[k][i];
                            arr[k][i] = '.';
                            break;
                        }
                    }
                }
            }
        }
    }
    return answer;
}