Programmers / Level 2 / 삼각 달팽이 / JS

< 문제 바로가기 >

프로그래머스

 

function solution(n) {
  let matrix = Array.from({ length: n }, () => new Array(n).fill(0));

  let currentNumber = 1;
  let [row, col] = [-1, 0];
  for (let i = 0; i < n; i += 1) {
    for (let j = i; j < n; j += 1) {
      if (i % 3 === 0) row += 1;
      if (i % 3 === 1) col += 1;
      if (i % 3 === 2) {
        row -= 1;
        col -= 1;
      }

      matrix[row][col] = currentNumber;
      currentNumber += 1;
    }
  }

  return matrix.flat().filter((el) => el !== 0);
}

const n = 4;
const result = solution(n);
console.log(result);

---

이 문제를 풀기 위해서 그림판을 난장판으로 만들었다...

 

간단히 말하면 결국 패턴이 존재하는데, 그 패턴안에 3개의 단계가 존재한다.

피라미드로 2차원 배열로 볼 때,

1. 아래로 내려가면서, 즉 행이 하나씩 더해지면서 값이 1이 증가하는 단계
2. 오른쪽으로 가면서, 즉 열이 하나씩 더해지면서 값이 1이 증가하는 단계
3. 좌상단 대각선 방향으로 가면서, 즉 행과 열이 하나씩 감소하면서 값이 1이 증가하는 단계

이 세가지 단계가 계속 반복이 된다.

 

따라서 3으로 나눴을때 나머지가 0이라면 1단계를, 1이라면 2단계를, 2라면 3단계를 수행해주면 된다.

 

모든 과정을 수행하고 나면 처음에 2차원 배열을 만들 때 모든 값을 0으로 초기화를 해놔서 중간중간에 0의 값이 껴있을텐데, 일단 flat()으로 2차원 배열을 1차원 배열로 차원을 하나 줄여준다. 그 후 filter()를 이용해 0이 아닌값들만 return 해주었다.