[프로그래머스] 기둥과 보 설치

문제

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

코딩테스트 연습 - 기둥과 보 설치

문제해결방법

1. 기둥과 보를 설치할 때, 설치할 수 있는 조건을 만족하는지 확인하고 설치한다.

2. 구조물을 제거할 때는 영향을 받는 구조물들이 유지될 수 있는지 확인 후 제거한다.

코드

package Programmers;
 
import java.util.ArrayList;
import java.util.List;
 
public class Solution_기둥과보설치 {
    
    public static boolean checkPillar(int x, int y, int[][] pillar, int[][] bo) {
        boolean check = false;
        if(y==0) check = true;
        if(y!=0 && pillar[y-1][x] == 1) check = true;
        if(bo[y][x]==1 || (x!=0 && bo[y][x-1]==1)) check = true;
        return check;
    }
    
    public static boolean checkBo(int x, int y, int[][] pillar, int[][] bo) {
        boolean check = false;
        if((y!=0 && (pillar[y-1][x]==1) || pillar[y-1][x+1]==1)) check = true;
        if(bo[y][x+1]==1 && (x!=0 && bo[y][x-1]==1)) check = true;
        return check;
    }
    
    public static boolean deletePillar(int x, int y, int[][] pillar, int[][] bo) {
        boolean check = true;
        pillar[y][x] = 0;
        if(pillar[y+1][x]==1) check &= checkPillar(x, y+1, pillar, bo);
        if(bo[y+1][x]==1) check &= checkBo(x, y+1, pillar, bo);
        if(x!=0 && bo[y+1][x-1]==1) check &= checkBo(x-1, y+1, pillar, bo);
        pillar[y][x] = 1;
        return check;
    }
    
    public static boolean deleteBo(int x, int y, int[][] pillar, int[][] bo) {
        boolean check = true;
        bo[y][x] = 0;
        if(bo[y][x+1]==1) check &= checkBo(x+1, y, pillar, bo);
        if(x!=0 && bo[y][x-1]==1) check &= checkBo(x-1, y, pillar, bo);
        if(pillar[y][x]==1) check &= checkPillar(x, y, pillar, bo);
        if(pillar[y][x+1]==1) check &= checkPillar(x+1, y, pillar, bo);
        bo[y][x] = 1;
        return check;
    }
    
    public static int[][] solution(int n, int[][] build_frame) {
        int[][] pillar = new int[n+2][n+2];
        int[][] bo = new int[n+2][n+2];
        for(int i=0; i<build_frame.length; i++) {
            int x = build_frame[i][0];
            int y = build_frame[i][1];
            int a = build_frame[i][2];
            int b = build_frame[i][3];
            if(b==1) {
                if(a==0 && checkPillar(x, y, pillar, bo)) pillar[y][x] = 1;
                else if(a==1 && checkBo(x, y, pillar, bo)) bo[y][x] = 1;
            } else {
                if(a==0 && deletePillar(x, y, pillar, bo)) pillar[y][x] = 0;
                else if(a==1 && deleteBo(x, y, pillar, bo)) bo[y][x] = 0;
            }
        }
        List<int[]> list = new ArrayList<>();
        for(int i=0; i<n+2; i++) {
            for(int j=0; j<n+2; j++) {
                if(pillar[j][i]==1) list.add(new int[] {i, j, 0});
                if(bo[j][i]==1) list.add(new int[] {i, j, 1});
            }
        }
        
        int[][] answer = new int[list.size()][3];
        for(int i=0; i<list.size(); i++) {
            for(int j=0; j<list.get(i).length; j++) {
                int[] temp = list.get(i);
                answer[i][j] = temp[j];
            }
        }
        return answer;
    }
    
    public static void main(String[] args) {
        int n = 5;
        int[][] build_frame = {{0,0,0,1},{2,0,0,1},{4,0,0,1},{0,1,1,1},{1,1,1,1},{2,1,1,1},{3,1,1,1},{2,0,0,0}, {1,1,1,0}, {2,2,0,1}};
        int[][] answer = solution(n, build_frame);
        for(int i=0; i<answer.length; i++) {
            for(int j=0; j<answer[i].length; j++) {
                System.out.print(answer[i][j]+" ");
            }
            System.out.println();
        }
    }
}

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