[프로그래머스]삼각 달팽이[JAVA, 자바]

문제 설명

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

풀이 방법

• 규칙을 파악해서 빡구현하는 문제였다.
• 먼저 삼각형의 규칙은 ↓ → ↖︎ 로 숫자가 증가하는 것을 볼 수 있다.
• 이를 통해 2차원 배열을 생성해서 증가하는대로 값을 넣어주면 된다.

알아야되는 개념

• 반복문
• 배열

class Solution {
    public int[] solution(int n) {
        int[] answer = {};
        int[][] arr = new int[n][n];
        int x = -1;
        int y = 0;
        int cnt = 0;
        for(int i=0;i<n;i++){
            for(int j=i;j<n;j++){
                if(i % 3 == 0) //아래
                    x++;
                else if(i % 3 ==1 ) { //오른쪽
                    y++;
                }
                else if(i % 3 == 2){ //대각선
                    x--;
                    y--;
                }
                arr[x][y] = ++cnt;
            }
        }
        answer = new int[cnt];
        int k = 0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(arr[i][j]==0) break;
                answer[k++] = arr[i][j];
            }
        }
        return answer;
    }
}