[Lv2] 삼각 달팽이

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

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

문제 설명

정수 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

• 문제 예시와 같습니다.

---

public int[] solution(int n) {
    int[][] triangle = new int[n][];
    for (int i = 0; i < n; ++i) {
        triangle[i] = new int[i + 1];
        triangle[i][0] = i + 1;
    }
    for (int i = 1; i < n; ++i) {
        triangle[n - 1][i] = n + i;
    }
    int number = n * 2;
    int end = 0;
    for (int i = n; i > 0; --i) {
        end += i;
    }
    int y = n - 1;
    int x = y;
    while (number <= end) {
        while (true) {
            if (triangle[y - 1][x - 1] != 0) {
                break;
            }
            --y;
            --x;
            triangle[y][x] = number++;
        }
        if (number > end) {
            break;
        }
        while (true) {
            if (triangle[y + 1][x] != 0) {
                break;
            }
            ++y;
            triangle[y][x] = number++;
        }
        if (number > end) {
            break;
        }
        while (true) {
            if (triangle[y][x + 1] != 0) {
                break;
            }
            ++x;
            triangle[y][x] = number++;
        }
    }
    int[] answer = new int[end];
    int index = 0;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j <= i; ++j) {
            answer[index++] = triangle[i][j];
        }
    }
    return answer;
}