[백준 / node.js] 1065번: 한수

출처: 백준

1065번: 한수

문제

어떤 양의 정수 X의 각 자리가 등차수열을 이룬다면, 그 수를 한수라고 한다. 등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다. N이 주어졌을 때, 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력하는 프로그램을 작성하시오. 

입력

첫째 줄에 1,000보다 작거나 같은 자연수 N이 주어진다.

출력

첫째 줄에 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력한다.

예제 입력 1

110

예제 출력 1

99

예제 입력 2

1

예제 출력 2

1

예제 입력 3

210

예제 출력 3

105

예제 입력 4

1000

예제 출력 4

144

내 코드

const readline = require("readline");
const rl = readline.createInterface({
    input : process.stdin,
    output : process.stdout
});

let input;

rl.on("line", function(line){
    input = Number(line);
    rl.close();
}).on("close", function(){
    let numbers = [];   // 100부터 입력받은 수 까지 담을 배열
    let arr = [];   // numbers에 속한 배열의 각 자리의 차를 담을 배열
    let count = 0;  // 한수 카운트

    if(input<100){  
        count = input;
    }else overHundred();
    
    function overHundred(){
        count = 99;
        for(let j=100; j<=input;j++){   // 100부터 입력받은 수까지 각 자리를 잘라 배열에 담기
            numbers.push(j.toString().split('').map(Number));
        }
        //ex) input이 210일 때
        // [ [ 1, 0, 0 ], [ 1, 0, 1 ], [ 1, 0, 2 ] . . .[ 2, 0, 9 ] [ 2, 1, 0 ] ]

        for(let i=0; i<numbers.length; i++){
            arr=[];
            for(let j=0; j<numbers[i].length-1; j++){   // numbers의 첫 번째 배열부터 각 자리의 차이 구하기
                arr.push((numbers[i][j]-numbers[i][j+1]));
            }
            /* 
            [ 1, 0 ]
            [ 1, -1 ]
            [ 1, -2 ]
            .
            .
            [ 2, -9 ]
            [ 1, 1 ] 
            */
            for(let k=1; k<arr.length; k++){    // 각 자리의 차이가 같을 경우 카운트
                if(arr[0] === arr[k]){
                    count++;
                }
            }
        }
    }
        
    console.log(count);
    process.exit();
})

결과

> 110 99

> 1 1

> 210 105

> 1000 144

 

코드가 깔끔하진 않은 것 같은데 나름 고민해서 풀어보았다.

이 문제를 풀 때 헷갈렸던 부분이 있는데 '100 미만인 수는 한수인 것인가?'이다. 예제 1번을 보면 알 수 있었다. 

풀이 방법은 210을 입력받았다고 가정할 때 100부터 210까지 각 자리 수를 배열로 담는다.

그럼

[ [ 1, 0, 0 ], [ 1, 0, 1 ], [ 1, 0, 2 ] . . .[ 2, 0, 9 ] [ 2, 1, 0 ] ] 이런 배열이 된다.

그 후에 각 배열들의 숫자의 차이를 구했다

[ 1, 0 ] [ 1, -1 ] [ 1, -2 ] . . . [ 2, -9 ] [ 1, 1 ]

각 숫자가 같으면 등차수열이기 때문에 숫자를 비교해서 한수이면 카운트를 셌다.

 

코드를 작성하면서도 좀 주절주절 쓰는 느낌이었는데 나중에 다시 풀어봐야겠다.