[프로그래머스/python] 다음 큰 숫자

문제 설명

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

 

입출력 예

n	result
78	83
15	23

 

코드

def change(num):
    num_list = []
    while num > 0:
        num_list.append(num % 2)
        num = num // 2
    return num_list

def solution(n):
    answer = 0
    b = n
    a = change(n)
    
    while True:
        b += 1
        a_1 = change(b)
        if a.count(1) == a_1.count(1):
            answer = b
            break
    
    return answer

 

풀이 방법

  • 십진수를 이진수로 전환해주는 change 함수를 만들었다.

def change(num):
    num_list = []
    while num > 0:
        num_list.append(num % 2)
        num = num // 2
    return num_list

  • b에 n(입력 값)을 저장하고 a에는 n의 이진수를 저장한 후, 무한루프를 이용해 b값을 1씩 증가시키고 a_1에 이진수로 전환된 b를 저장

  • a의 1의 개수와 a_1의 1의 개수가 같을 시 b를 반환한다.