오늘 한 일

• 알고리즘 코드카타
  • 콜라츠 수열 만들기

알고리즘 코드카타

콜라츠 수열 만들기

문제

모든 자연수 x에 대해서 현재 값이 x이면 x가 짝수일 때는 2로 나누고, x가 홀수일 때는 3 * x + 1로 바꾸는 계산을 계속해서 반복하면 언젠가는 반드시 x가 1이 되는지 묻는 문제를 콜라츠 문제라고 부릅니다.

그리고 위 과정에서 거쳐간 모든 수를 기록한 수열을 콜라츠 수열이라고 부릅니다.

계산 결과 1,000 보다 작거나 같은 수에 대해서는 전부 언젠가 1에 도달한다는 것이 알려져 있습니다.

임의의 1,000 보다 작거나 같은 양의 정수 n이 주어질 때 초기값이 n인 콜라츠 수열을 return 하는 solution 함수를 완성해 주세요.

풀이

function solution(n) {
  function collatz(n) {
    if (n % 2 === 0) {
      return n / 2;
    } else {
      return 3 * n + 1;
    }
  }

  const result = [];
  while (n !== 1) {
    result.push(n);
    n = collatz(n);
  }
  result.push(1);
  return result;
}

함수 내부에 새로운 함수를 만들어서 (밖에 두면 왜인지 프로그래머스에서 오류가 나 함수 내에 넣었다.) 짝수 및 홀수를 판단하여 n을 변경시켜 return을 시킨다. n이 1이 될 때까지 반복하여 result에 n을 push하고 n을 아까 만든 함수를 통해 변경시킨다. n이 1이 되었을 경우 반복문이 끝나고 마지막 숫자가 될 1을 result에 push해준 다음 result를 return시킨다.

다른 사람의 풀이

function solution(n, arr = []) {
    arr.push(n)
    if (n === 1) return arr
    if (n % 2 === 0) return solution(n / 2, arr)
    return solution(3 * n + 1, arr)
}

하나의 함수로 재귀시키는 식을 알게 되었다. 확실히 훨씬 더 간단하고 보기 좋은 코드라 가져왔다.