[알고리즘] 삼각 달팽이

문제 설명

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

제한사항

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

입출력 예

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]

나의 풀이

function solution(n) {
    var answer = new Array(n).fill(0).map(() => new Array(n).fill(0));
    let index = 0; 
    let end = 0;
    const x = [0, 1, -1];
    const y = [1, 0, -1];
    let position = [0,0];
    
    answer[0][0] = 1;
    
    for(let i = 1; i <= n; i++) {
        end += i;
    }
    
    for(let i = 2; i<= end; i++) {
        position[0] = position[0] + y[index % 3];
        position[1] = position[1] + x[index % 3];
        
        answer[position[0]][position[1]] = i;
        
        const nextY = position[0] + y[index % 3];
        const nextX = position[1] + x[index % 3];
        
        if(!answer[nextY] || answer[nextY] && answer[nextY][nextX] !== 0) {
            index += 1;
        }
    }
    
    let realAnswer = []
    
    for(let i = 0; i < answer.length; i++) { 
        const current = answer[i].filter((data) => data !== 0);
        
        realAnswer = realAnswer.concat(current);
    }
    
    return realAnswer;
}