[프로그래머스/C++] [1차] 프렌즈4블록

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

코딩테스트 연습 - [1차] 프렌즈4블록

[1차] 프렌즈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가 사용된다.

출력 형식

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

 

풀이

입출력 예시

 

풀이

생각하는 그대로 문제에 반영하면 되었다.

1. 2x2 크기의 판별 매트릭스를 한번씩 돌아가며 지워야할 블록들을 찾는다.

2. 만약 지워야할 블록이 더 이상 존재하지 않으면 값을 반환한다.

3. 아니라면 블록들을 떨어뜨린다.

3-1. 지워질 블록들은 '#'으로 표기한다.

3-2. '#'인 블록들을 맨 위로 올린다.

4. 더 이상 지워야할 블록이 없을 때 까지 재귀적으로 반복한다.

 

int check_board(int m, int n, vector<string> board) {
    set<vector<int>> check;
    
    /*for (const auto& b : board) {
        cout << b << endl;
    }
    cout << endl;*/
    
    // 2x2 매트릭스로 찾음
    for (int i = 0; i < m - 1; i++) {
        for (int j = 0; j < n - 1; j++) {
            if (board[i][j] != '#' && board[i][j] == board[i + 1][j]) {
                if (board[i][j + 1] == board[i + 1][j + 1]) {
                    if (board[i][j] == board[i + 1][j + 1]) {
                        check.insert({i, j});
                        check.insert({i, j + 1});
                        check.insert({i + 1, j});
                        check.insert({i + 1, j + 1});
                    }
                }
            }
        }
    }
    
    if (check.empty()) {
        return 0;
    }
    else {
        drop_block(board, check);
        return check.size() + check_board(m, n, board);
    }
}

check는 지워질 블록들을 저장하는 set이다. set으로 하는 이유는 겹쳐지는 부분도 있기 때문이다.

---

void drop_block(vector<string>& board, set<vector<int>> check) {
    for (const auto& idx : check) {
        board[idx[0]][idx[1]] = '#';
        for (int i = idx[0]; i > 0; i--) {
            swap(board[i][idx[1]], board[i - 1][idx[1]]);
        }
    }
}

check에 있는 좌표를 통해 떨어뜨려야 할 블럭을 받아와서, '#'으로 바꿔주는 처리를 해준다. 그 다음, 위에서 아래로 내려준다.

---

전체 코드

#include <string>
#include <vector>
#include <set>
#include <algorithm>
#include <iostream>

using namespace std;

int check_board(vector<string> board);
void drop_block(vector<string>& board, set<vector<int>> check);

int check_board(int m, int n, vector<string> board) {
    set<vector<int>> check;
    
    /*for (const auto& b : board) {
        cout << b << endl;
    }
    cout << endl;*/
    
    // 2x2 매트릭스로 찾음
    for (int i = 0; i < m - 1; i++) {
        for (int j = 0; j < n - 1; j++) {
            if (board[i][j] != '#' && board[i][j] == board[i + 1][j]) {
                if (board[i][j + 1] == board[i + 1][j + 1]) {
                    if (board[i][j] == board[i + 1][j + 1]) {
                        check.insert({i, j});
                        check.insert({i, j + 1});
                        check.insert({i + 1, j});
                        check.insert({i + 1, j + 1});
                    }
                }
            }
        }
    }
    
    if (check.empty()) {
        return 0;
    }
    else {
        drop_block(board, check);
        return check.size() + check_board(m, n, board);
    }
}

void drop_block(vector<string>& board, set<vector<int>> check) {
    for (const auto& idx : check) {
        board[idx[0]][idx[1]] = '#';
        for (int i = idx[0]; i > 0; i--) {
            swap(board[i][idx[1]], board[i - 1][idx[1]]);
        }
    }
}

int solution(int m, int n, vector<string> board) {
    int answer = 0;

    answer = check_board(m, n, board);
    
    return answer;
}

 

결과