프로그래머스 삼각 달팽이

프로그래머스 바로가기

문제 설명

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

접근 방식

방향은 총 3방향이 있고 조건에 맞게 방향을 바꿔주면 됩니다.

코드

import java.util.*;

class Solution {
    static int[] dx = {1, 0, -1};
    static int[] dy = {0, 1, -1};
    public List<Integer> solution(int n) {
        List<Integer> answer = new ArrayList<>();
        
        int[][] arr = new int[n+2][n+2];

        int d = 0;
        int count = 1;
        int x = 1;
        int y = 1;
        
        if(n == 1) answer.add(1);
        else{
            while(true){
                arr[x][y] = count++;

                int nx = x + dx[d];
                int ny = y + dy[d];

                if(nx > n || ny > n || arr[nx][ny] != 0){
                    d = (d + 1) % 3;

                    x += dx[d];
                    y += dy[d];

                    if(arr[x][y] != 0) break;
                }else{
                    x = nx;
                    y = ny;
                }
            
            }
            for(int i = 1; i <= n ; i++){
                for(int j = 1; j <= i; j++){
                    answer.add(arr[i][j]);   
                }
            }
        }
        return answer;
    }
}

Disscussion

어렵지 않은 달팽이 문제였습니다.