[프로그래머스] - [1차] 프렌즈4블록(Kotlin)

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

프로그래머스

문제

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

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

board map

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

board map

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

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

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가 사용된다.
출력 형식
입력으로 주어진 판 정보를 가지고 몇 개의 블록이 지워질지 출력하라.

 

풀이(Kotlin)

class Solution {
    fun solution(
        n: Int,//폭,board[0].length
        m: Int,//높이,board.size
        board: Array<String>,
    ): Int {
        val charArrays = board.map { it.toCharArray() }
        while (true) {
            val arrayList: ArrayList<Pair<Int, Int>> = arrayListOf()
            for (i in 0 until m - 1) {
                for (j in 0 until n - 1) {
                    if (charArrays[i][j] != '0' && charArrays[i][j] == charArrays[i][j + 1] && charArrays[i][j] == charArrays[i + 1][j] && charArrays[i][j] == charArrays[i + 1][j + 1]) {
                        arrayList.add(Pair(i, j))
                    }
                }
            }
            if (arrayList.isEmpty()) break
            arrayList.forEach { pair ->
                charArrays[pair.first][pair.second] = '0'
                charArrays[pair.first + 1][pair.second] = '0'
                charArrays[pair.first][pair.second + 1] = '0'
                charArrays[pair.first + 1][pair.second + 1] = '0'
            }
            for (i in 0 until n) {
                for (j in 1 until m) {
                    if (charArrays[j][i] == '0' && charArrays[j - 1][i] != '0') {
                        for (k in j downTo 1) {
                            if (charArrays[k-1][i] == '0'){
                                charArrays[k][i] = '0'
                                break
                            }
                            charArrays[k][i] = charArrays[k - 1][i]
                        }

                        if (charArrays[0][i] != '0') charArrays[0][i] = '0'
                    }
                }
            }
        }
        return charArrays.fold(0) { acc, charArray -> acc + charArray.count { it == '0' } }
    }
}

• 해당문제는 배열을 잘이용하면 손쉽게 풀립니다.
• 배열을 이용해서 2*2칸이 있는지 검사합니다. 없다면 멈추고 남은 배열에서 0인 칸을 카운트하면 됩니다.
• 있을 경우 해당 구역의 좌표를 저장했다가 순환 후 저장한 구역의 좌표들로 판을 비우는 작업을 합니다.
• 비웠다면 반복문을통해 위에 있는 값이 떨어지는 작업을 합니다.
• 이러한 로직을 반복하여 2*2칸이 없을때까지 반복합니다.

풀이(Java)

import java.util.ArrayList;
class Solution {
    public int solution(int m, int n, String[] board) {
        char[][] charArrays = new char[m][n];
        for (int i = 0; i < m; i++) {
            charArrays[i] = board[i].toCharArray();
        }

        while (true) {
            ArrayList<Pair> arrayList = new ArrayList<>();

            for (int i = 0; i < m - 1; i++) {
                for (int j = 0; j < n - 1; j++) {
                    if (charArrays[i][j] != '0' && charArrays[i][j] == charArrays[i][j + 1] &&
                            charArrays[i][j] == charArrays[i + 1][j] && charArrays[i][j] == charArrays[i + 1][j + 1]) {
                        arrayList.add(new Pair(i, j));
                    }
                }
            }
            if (arrayList.isEmpty()) {
                break;
            }
            for (Pair pair : arrayList) {
                charArrays[pair.first][pair.second] = '0';
                charArrays[pair.first + 1][pair.second] = '0';
                charArrays[pair.first][pair.second + 1] = '0';
                charArrays[pair.first + 1][pair.second + 1] = '0';
            }
            for (int i = 0; i < n; i++) {
                for (int j = 1; j < m; j++) {
                    if (charArrays[j][i] == '0' && charArrays[j - 1][i] != '0') {
                        
                        for (int k = j; k > 0; k--) {
                            if (charArrays[k-1][i] == '0') {
                                charArrays[k][i] = '0';
                                break;
                            }
                            charArrays[k][i] = charArrays[k - 1][i];
                        }
                        if (charArrays[0][i] != '0') charArrays[0][i] = '0';
                    }
                }
            }
        }

        int count = 0;
        for (char[] charArray : charArrays) {
            for (char c : charArray) {
                if (c == '0') {
                    count++;
                }
            }
        }
        return count;
    }

    private class Pair {
        int first;
        int second;

        Pair(int first, int second) {
            this.first = first;
            this.second = second;
        }
    }
}

• 해당문제는 Kotlin 풀이를 받지 못하기 때문에 자바로 변환하여 제출하였습니다.

결과