프로그래머스 월간 코드 챌린지 시즌1 LV2 - 삼각 달팽이

문제 설명

정수 n이 매개변수로 주어집니다. 다음 그림과 같이 밑변의 길이와 높이가 n인 삼각형에서 맨 위 꼭짓점부터 반시계 방향으로 달팽이 채우기를 진행한 후, 첫 행부터 마지막 행까지 모두 순서대로 합친 새로운 배열을 return 하도록 solution 함수를 완성해주세요.

---

제한사항

• n은 1 이상 1,000 이하입니다.

---

입출력 예

n         result

4	[1,2,9,3,10,8,4,5,6,7]
5	[1,2,12,3,13,11,4,14,15,10,5,6,7,8,9]
6	[1,2,15,3,16,14,4,17,21,13,5,18,19,20,12,6,7,8,9,10,11]

---

입출력 예 설명

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 문제 예시와 같습니다.

입출력 예 #3

• 문제 예시와 같습니다.

---

//0 1 3 6 7 8 9 5 2 4
import java.util.Arrays;
class Solution {
    int[] answer;
    public int[] solution(int ns) {
        int num = 0;
        for(int i=1; i<=ns; i++){
            num+=i;
        }
        answer = new int[num];
        int target = 0;
        int plusidx = 1;
        int n = 1;
        trisnail(0,1,1,ns);
        return answer;
    }
    
    void trisnail(int startidx, int n, int cnt, int ns){
        int target = startidx;
        int plusidx = cnt; 
        int count = 0;
        while(true){
            if(target>=answer.length) return;
            answer[target] = n;
            target+=plusidx;
            n++; plusidx++;
            if(n>answer.length) return;
            if(plusidx==ns){
                while(answer[target]==0){
                    answer[target] = n;
                    n++; target++; 
                    if(target==answer.length) {target--; break;}
                    else if(answer[target]!=0){
                        target--;break;
                    }
                }
                if(n>answer.length) return;
                int tcnt=0;
                while(answer[target-plusidx]==0){
                    target-=plusidx; 
                    answer[target] = n;
                    n++; plusidx--;
                    if(answer[target-plusidx]!=0){
                        break;
                    }
                }
                break;
            }
        }
        cnt= getcnt(ns,target);
        trisnail(target,n-1,cnt,ns-1);
    }
    int getcnt(int num, int target){
        int a = 0;
        for(int i=1; i<=num; i++){
            a+=i;
            if(a>=target){
                return i;
            }
        }
        return 0;
    }
}