프로그래머스 Lv.2 - 다음 큰 숫자

문제 설명

자연수 n이 주어졌을 때, n의 다음 큰 숫자는 다음과 같이 정의 합니다.

조건 1. n의 다음 큰 숫자는 n보다 큰 자연수 입니다.
조건 2. n의 다음 큰 숫자와 n은 2진수로 변환했을 때 1의 갯수가 같습니다.
조건 3. n의 다음 큰 숫자는 조건 1, 2를 만족하는 수 중 가장 작은 수 입니다.
예를 들어서 78(1001110)의 다음 큰 숫자는 83(1010011)입니다.

자연수 n이 매개변수로 주어질 때, n의 다음 큰 숫자를 return 하는 solution 함수를 완성해주세요.

제한 사항

• n은 1,000,000 이하의 자연수 입니다.

입출력 예

접근방법

1. n을 이진수로 만든 뒤 count 변수를 선언해 1을 카운트한다. 2. 그리고 큰 숫자 중 1의 갯수가 같은 숫자를 찾는다. 방법은 심플한데 반복되는 코드가 너무 많았다. 그러다 문득 든 생각, 1번의 방법을 함수로 만들어서 2번에서 사용하면? 예상대로 코드가 굉장히 짧아졌다!

2. n의 이진수에서 1을 카운트해줘야되므로 countOne 생성한 후 함수를 만든다. n을 받아서 이진수로 만들고, 이진수를 돌면서 1을 카운트해준 뒤 카운트(1의 갯수)를 리턴한다.

3. 다음 숫자를 알아야되는데, 똑같이 진행되므로 nextCount 변수를 생성한 후 countOne함수를 호출한다. 즉, nextCount에는 1번의 카운트의 값이 저장된다.

4. 다음 큰 숫자를 찾아내기 위해서 for문을 생성해준다. n+1 부터 진행하면되므로(n 다음 큰 숫자이므로) for문의 i는 n+1, 두번째 조건은 n은 1,000,000 이하 자연수이므로 i<=1000001 세번째 조건은 i++ i는 계속 증가한다.

5. 이렇게 돌면서 countOne(n)의 이진수 1의 갯수와 nextCount(i)의 이진수 1의 갯수가 맞을 때 i를 리턴한다.

코드

function solution(n) {
    const countOne = (n) => {
        let binary = n.toString(2);
        let count = 0;
        for (let i = 0; i < binary.length; i++) {
            if (binary[i] === '1') {
                count++;
            }
        }
        return count;
    }

    const nextCount = countOne(n); 
    for (let i=n+1; i<=1000001; i++) {
        if (countOne(i) === nextCount) {
            return i;
        }
    }
}