백준 JS | 4673번 셀프 넘버

출처 : 백준 온라인저지

https://www.acmicpc.net/problem/4673

4673번: 셀프 넘버

문제요약

양의 정수 n에 대해서 d(n)을 n과 n의 각 자리수를 더하는 함수라고 정의하자. 예를 들어, d(75) = 75+7+5 = 87이다. 양의 정수 n이 주어졌을 때, 이 수를 시작해서 n, d(n), d(d(n)), d(d(d(n))), ...과 같은 무한 수열을 만들 수 있다.

 

예를 들어, 33으로 시작한다면 다음 수는 33 + 3 + 3 = 39이고, 그 다음 수는 39 + 3 + 9 = 51, 다음 수는 51 + 5 + 1 = 57이다. 이런식으로 다음과 같은 수열을 만들 수 있다.

33, 39, 51, 57, 69, 84, 96, 111, 114, 120, 123, 129, 141, ...

n을 d(n)의 생성자라고 한다. 위의 수열에서 33은 39의 생성자이고, 39는 51의 생성자, 51은 57의 생성자이다. 생성자가 한 개보다 많은 경우도 있다. 예를 들어, 101은 생성자가 2개(91과 100) 있다. 

 

생성자가 없는 숫자를 셀프 넘버라고 한다. 100보다 작은 셀프 넘버는 총 13개가 있다. 1, 3, 5, 7, 9, 20, 31, 42, 53, 64, 75, 86, 97

10000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 출력하는 프로그램을 작성하시오.

나의 답안 - 정답

const arr = [];
const selfNum = [];
let num = 0;

// 1~10000 까지의 각 숫자가 어느 숫자의 생성자인지 계산하고
// 해당 생성자들로 이루어진 배열을 만든다. 
for (let i = 1; i <= 10000; i++) {
  num += i;
  for (let j = 0; j < i.toString().length; j++) {
    num += +i.toString()[j];
  }
  arr.push(num);
  num = 0;
}

// 1~10000 까지의 숫자 중에서 생성자 배열에 포함되지 않은 숫자들로 새로운 배열을 만든다.
for (let k = 1; k <= 10000; k++) {
  if (arr.indexOf(k) === -1) {
    selfNum.push(k);
  }
}

// 새로 만들어진 배열은 생성자 배열에 포함되지 않는, 즉 셀프 넘버들로 이루어져있다. 
// 출력하면 정답
for (let m = 0; m < selfNum.length; m++) {
  console.log(selfNum[m]);
}​

답은 맞았지만 for 문을 세번이나 써서인지 메모리를 너무 많이 잡아먹었다. 

이를 개선하기 위해 새로운 배열을 다시 for문을 돌리지 않고 바로 출력되도록 아래와 같이 for문 2개로 구현했는데,

오히려 이게 더 많은 메모리를 차지했다. 

3분 전 제출한 답안이 아래의 코드.. ( for문을 3개 -> 2개로 줄였음에도 오히려 메모리는 더 차지한다.)

const arr = [];
let num = 0;

for (let i = 1; i <= 10000; i++) {
  num += i;
  for (let j = 0; j < i.toString().length; j++) {
    num += +i.toString()[j];
  }
  arr.push(num);
  num = 0;
}

for (let k = 1; k <= 10000; k++) {
  if (arr.indexOf(k) === -1) {
    console.log(k);
  }
}

 

아무래도 콘솔을 여러번 출력하는 것이 원인이 아닌가 싶다. ( 난이도가 낮은 문제에서 console을 여러번 출력하면 효율성 테스트에서 막히곤 했다. 콘솔이 생각보다 메모리 소모가 크다고 한다. )

const arr = [];
let selfNum = "";
let num = 0;

for (let i = 1; i <= 10000; i++) {
  num += i;
  for (let j = 0; j < i.toString().length; j++) {
    num += +i.toString()[j];
  }
  arr.push(num);
  num = 0;
}

for (let k = 1; k <= 10000; k++) {
  if (arr.indexOf(k) === -1) {
    selfNum += k + "\n";
  }
}

console.log(selfNum);

배열에 담아 맵을 돌리지 않고 셀프넘버들을 string 형태로 저장한 후 한번의 console 로 출력하니 10% 이상 메모리 소모가 개선되었다.

 

다른 사람의 답안

다른 사람들의 답안을 살펴보던 중 압도적으로 메모리 소모와 연산속도가 빠른 답안을 찾았다. (jsoonee)

let arr = Array(10001).fill(false);
let n = 1;
while (n<10000) {
    arr[n+(n+'').split('').reduce((a,c)=>a+ +c,0)]=true;
    n++;
}
let r = [];
arr.forEach((v,i) => {
    if (!v) r.push(i);
})
console.log(r.slice(1).join('\n'));

while문과 forEach를 사용해서 풀이를 했는데, 잘 이해가 되지 않아 추후에 더 살펴봐야할 것 같다.