프로그래머스 LV2. 프렌즈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의 경우, 첫 번째에는 A 블록 6개가 지워지고, 두 번째에는 B 블록 4개와 C 블록 4개가 지워져, 모두 14개의 블록이 지워진다.
• 입출력 예제 2는 본문 설명에 있는 그림을 옮긴 것이다. 11개와 4개의 블록이 차례로 지워지며, 모두 15개의 블록이 지워진다.

 

코딩테스트 연습 - [1차] 프렌즈4블록 | 프로그래머스 (programmers.co.kr)

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

 

😊 나의 코드

import java.util.*;
class Solution {
    public int solution(int m, int n, String[] board) {
       int answer = 0;
        ArrayList[] arrList = new ArrayList[n];
        for (int i = 0; i < n; i++) {
            ArrayList<Character> arr = new ArrayList();
            for (int j = m - 1; j >= 0; j--) {
                arr.add(board[j].charAt(i));
            }
            arrList[i] = arr;
        }

        boolean deleteFlg = true; //삭제할 블록이 있는지 여부
        while (deleteFlg) {
            deleteFlg = false;
            TreeSet<Integer>[] setList = new TreeSet[n]; //삭제할 블록 리스트
            for (int i = 0; i < n - 1; i++) {
                ArrayList<Character> arr = arrList[i];
                ArrayList<Character> nextArr = arrList[i + 1];
                TreeSet<Integer> deleteBlock = new TreeSet<>(Collections.reverseOrder());
                TreeSet<Integer> nextDeleteBlock = new TreeSet<>(Collections.reverseOrder());
                if (i > 0) {
                    deleteBlock = setList[i];
                }
                for (int j = 0; j < arr.size() - 1 && j < nextArr.size() - 1; j++) {
                    if (arr.get(j) == arr.get(j + 1)) {
                        if (nextArr.get(j) == arr.get(j) && nextArr.get(j) == nextArr.get(j + 1)) {
                            deleteFlg = true; //삭제할 블록이 있을때
                            deleteBlock.add(j + 1);
                            deleteBlock.add(j);
                            nextDeleteBlock.add(j + 1);
                            nextDeleteBlock.add(j);
                        }
                    }
                }
                setList[i] = deleteBlock;
                setList[i + 1] = nextDeleteBlock;
            }

            if (deleteFlg) { //삭제할 블록이 있다면 블록 삭제
                for (int i = 0; i < n; i++) {
                    TreeSet<Integer> deleteBlock = setList[i];
                    ArrayList<Character> arr = arrList[i];
                    for (int d : deleteBlock) {
                        answer++;
                        arr.remove(d);
                    }
                    arrList[i] = arr;
                }
            }
        }

        return answer;
    }
}

 

세로 한줄한줄 씩 ArrayList에 블록을 담고 블록 리스트를 ArrayList[] arrList 배열에 담았다.

ArrayList를 쓴 이유는 차례로 블록이 지워지는게 아니라 중간에서 블록이 지워지기 때문에

arr.remove(지울블록 인덱스);을 사용해서 블록을 지우기 위해서였다.

그래서 블록을 맨 아래에서 위로 arr에 담았다.

(위에 블록이 비어있을 수도 있으니 아래부터 차례로 담음)

다른 풀이들을 보니 중간 블록이 지워졌을 때 위의 블록들을 한칸씩 아래로 옮기는 방법도 있었다.