프로그래머스 코딩테스트/Javascript

[프로그래머스] Javascript : 다음 큰 숫자

김콩콩쥐 2024. 2. 16. 11:37

문제 설명

자연수 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 이하의 자연수 입니다.

나의 풀이

function solution(n) {
    const binary = n.toString(2)
    const count = binary.match(/1/g).length
    let arr = []

    for (let i = n + 1; i < 2 ** ((n + 1).toString(2).length); i++) {
        arr.push(i.toString(2))
    }

    const newArr = arr.filter((el) => el.match(/1/g).length === count)

    return parseInt(newArr[0], 2)
}

n을 2진수로 변환 후 1의 갯수를 센다.
그리고 for문을 통해 새로운 배열에 2진수로 변환한 값을 삽입했다.
하지만 테스트1에서 실패, 효율성 테스트에선 모두 실패가 떴다.

최종 풀이

function solution(n) {
    const count = n.toString(2).replace(/0/gi,'').length
    let result = n + 1

    while (n < result) {
        const resultCount = result.toString(2).replace(/0/gi,'').length

        if (resultCount === count) {
            break;
        }
        result++
    }

    return result
}

2진수 전체에서 1을 세기보단 먼저 0을 삭제하고 1의 길이를 센다.
result 는 현재값으로 생각하면 되고 현재값의 1의 길이가 같다면 break로 반복문을 종료한다.

다른 사람의 풀이

function solution(n, a = n + 1) {
    return n.toString(2).match(/1/g).length === a.toString(2).match(/1/g).length ? a : solution(n, a + 1);
}

이건 재귀 알고리즘을 사용한 방법이다.