[Python] 프로그래머스 코딩테스트 연습 level2 (다음 큰 숫자)

1. 문제 설명

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

• 조건 1. n의 다음 큰 숫자는 n보다 큰 자연수 입니다.
• 조건 2. n의 다음 큰 숫자와 n은 2진수로 변환했을 때 1의 갯수가 같습니다.
• 조건 3. n의 다음 큰 숫자는 조건 1, 2를 만족하는 수 중 가장 작은 수 입니다.

예를 들어서 78(1001110)의 다음 큰 숫자는 83(1010011)입니다.

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

 

 

 

 

2. 제한 조건

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

 

 

 

 

 

 

3. 내 풀이

def solution(n):
    n1 = str(bin(n)).count('1')
    for i in range(n+1, 1000001):
        if str(bin(i)).count('1') == n1:
            break
    return i

 

1. bin()함수를 사용하여 10진수를 이진수로 변환한 뒤 count()함수로 1의 개수 카운트

2. 입력값 n보다 큰 숫자부터 최대 수인 1,000,000까지 반복하여 n과 1의 개수가 같으면 break 후 i 반환

 

 

 

4. 다른 사람 풀이

def nextBigNumber(n):
    num1 = bin(n).count('1')
    while True:
        n = n + 1
        if num1 == bin(n).count('1'):
            break
    return n

 

방식은 비슷하지만 for문이 아닌 while문 사용