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

문제

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

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

문제해결방법

1. check 함수를 통해 보드를 돌면서 지워지는 블록들을 체크한다.

2. redraw 함수로 블록들을 지우고 아래로 내린다.

3. 위 과정을 더 이상 블록이 지워지지 않을때까지 계산한다

코드

package Programmers;
 
public class Solution_프렌즈4블록 {
 
    public static int[][] dir = {{1, 0}, {0, 1}, {1, 1}};
    
    public static void redraw(char[][] map, boolean[][] v) {
        int m = map.length;
        int n = map[0].length;
        for(int i=0; i<n; i++) {
            for(int j=m-1; j>=0; j--) {
                if(v[j][i]) {
                    v[j][i] = false;
                    char ch = '^';
                    for(int k=j-1; k>=0; k--) {
                        if(!v[k][i]) {
                            ch = map[k][i];
                            v[k][i] = true;
                            break;
                        }
                    }
                    map[j][i] = ch;
                }
            }
        }
    }
    
    public static int check(int i, int j, char[][] map, boolean[][] v) {
        int count = 0;
        char ch = map[i][j];
        for(int d=0; d<dir.length; d++) {
            int ii = i+dir[d][0];
            int jj = j+dir[d][1];
            if(ii>=0 && ii<map.length && jj>=0 && jj<map[0].length && map[ii][jj]==ch) continue;
            else return 0;
        }
        for(int d=0; d<dir.length; d++) {
            int ii = i+dir[d][0];
            int jj = j+dir[d][1];
            if(!v[ii][jj]) {
                v[ii][jj] = true;
                count++;
            }
        }
        if(!v[i][j]) {
            v[i][j] = true;
            count++;
        }
        return count;
    }
    
    public static int solution(int m, int n, String[] board) {
        int answer = 0;
        boolean[][] v = new boolean[m][n];
        char[][] map = new char[m][n];
        for(int i=0; i<m; i++) {
            for(int j=0; j<n; j++) map[i][j] = board[i].charAt(j);
        }
        
        while(true) {
            int sum = 0;
            for(int i=0; i<m; i++) {
                for(int j=0; j<n; j++) {
                    if(map[i][j]=='^') continue;
                    sum += check(i, j, map, v);
                }
            }
            if(sum==0) break;
            answer += sum;
            redraw(map, v);
        }
        return answer;
    }
    
    public static void main(String[] args) {
        int m = 6;
        int n = 6;
        String[] board = {"TTTANT", "RRFACC", "RRRFCC", "TRRRAA", "TTMMMF", "TMMTTJ"};
        System.out.println(solution(m, n, board));
    }
}

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.