[프로그래머스 Lv2] 삼각 달팽이(Java)

1. 문제

programmers.co.kr/learn/courses/30/lessons/68645?language=java

코딩테스트 연습 - 삼각 달팽이

2. 풀이

• 배열문제, 별찍기가 생각났다.

3. 코드

class Solution {
    public int[] solution(int n) {
        int max = getMax(n);
        int[] answer = new int[max];
        int[][] arr = new int[n][n];
        for(int i =0; i<n; i++){
            for(int j =0; j<=i; j++){
                arr[i][j] = 0;
            }
        }
        int r=0, c=0, k=1;
        arr[r][c] = k;

        while(k<max){
            while(r+1<n && arr[r+1][c]==0){
                arr[++r][c] = ++k;
            }
            while(c+1<n && arr[r][c+1]==0){
                arr[r][++c] = ++k;
            }
            while (r-1>0 && c-1>0 && arr[r-1][c-1]==0){
                arr[--r][--c] = ++k;
            }
        }
        int idx=0;
        for(int i =0; i<n; i++){
            for(int j =0; j<=i; j++){
                answer[idx++] = arr[i][j];
            }
        }
        return answer;
    }
       static int getMax(int n){
        if(n==1) return 1;
        return getMax(n-1)+n;
    }
}

 

반응형