개발 기초/알고리즘

[프로그래머스] LV2. 프렌즈4블록 (Java)

숩니따 2025. 3. 18. 16:13

문제

블라인드 공채를 통과한 신입 사원 라이언은 신규 게임 개발 업무를 맡게 되었다. 이번에 출시할 게임 제목은 "프렌즈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가 사용된다.

출력 형식

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

입출력 예제

m n board answer
4 5 ["CCBDE", "AAADE", "AAABF", "CCBBF"] 14
6 6 ["TTTANT", "RRFACC", "RRRFCC", "TRRRAA", "TTMMMF", "TMMTTJ"] 15

풀이

  1. M - 1, N - 1 까지 블록을 순회하면서 시계방향으로 자신과 같은 블록이 있는지 확인
  2. 같은 블록이 있다면 터뜨리고 빈 블록으로 바꾸기
  3. 아래에서 위 행으로 이동하면서 빈 블록이면 큐에 넣고 큐가 비지 않았다면 캐릭터 블록 아래로 내리기
  4. 터지는 블록이 있을 때까지 반복

사용 개념

Queue

for (int j = 0; j < n; j++) {
    Queue<Integer> queue = new LinkedList<>();
    // 행 밑에서부터 위로 거슬러올라가기
    for (int i = m - 1; i >= 0; i--) {
        if (arr[i][j] == '1') {
            // 1. 빈공간이 있다면 큐에 넣기
            queue.offer(i);
        } else {
		        // 2. 큐가 비지 않았다면 아래 빈 공간이 있다는 것으로
            if (!queue.isEmpty()) {
		            // 3. 큐에서 가장 밑에 있는 빈 공간을 꺼내고
                int idx = queue.poll();
                // 4. 해당 빈 공간에 위에 있는 캐릭터를 집어넣고
                arr[idx][j] = arr[i][j];
                // 5. 캐릭터가 있던 자리는 빈 공간으로 만들고 큐에 넣기
                arr[i][j] = '1';
                queue.offer(i);
            }
        }
  }
}

코드

import java.util.*;

class Solution {
    static char[][] arr;
    // 4개 겹친 블록 터뜨리는 함수
    private static int popBoard(int n, int m) {
        int[][] ds = {{0, 1}, {1, 0}, {1, 1}};  // 우, 하 ,우하 확인
        boolean[][] check = new boolean[m][n];  // 터뜨리는지 확인하는 배열
        for (int i = 0; i < m - 1; i++) {
            for (int j = 0; j < n - 1; j++) {
                char friend = arr[i][j];
                // 빈공간이면 넘어가기
                if (friend == '1') {
                    continue;
                }
                boolean flag = true;
                for (int[] d : ds) {
                    int nr = i + d[0];
                    int nc = j + d[1];
                    // 캐릭터가 같지 않다면 break
                    if (arr[nr][nc] != friend) {
                        flag = false;
                        break;
                    }
                }
                // 캐릭터가 같으면 check 배열에 표시
                if (flag) {
                    check[i][j] = true;
                    for (int[] d : ds) {
                        int nr = i + d[0];
                        int nc = j + d[1];
                        check[nr][nc] = true;
                    }
                }

            }
        }
        int count = 0;
        // 없어진 공간 빈공간으로 만들기
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (check[i][j]) {
                    arr[i][j] = '1';
                    count++;
                }
            }
        }
        return count;
    }
    // 빈 공간 내리는 함수
    private static void downBoard(int n, int m) {
        for (int j = 0; j < n; j++) {
            Queue<Integer> queue = new LinkedList<>();
            for (int i = m - 1; i >= 0; i--) {
                if (arr[i][j] == '1') {
                    queue.offer(i);
                } else {
                    if (!queue.isEmpty()) {
                        int idx = queue.poll();
                        arr[idx][j] = arr[i][j];
                        arr[i][j] = '1';
                        queue.offer(i);
                    }
                }
            }
        }
    }

    public int solution(int m, int n, String[] board) {
        arr = new char[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                arr[i][j] = board[i].charAt(j);
            }
        }
        int answer = 0;

        while (true) {
            int count = popBoard(n, m);
            if (count > 0) {
                answer += count;
                downBoard(n, m);
            } else {
                break;
            }
        }

        return answer;
    }
}