코딩테스트/Python

[Level 2] 다음 큰 숫자

eunki 2021. 10. 11. 10:34
728x90

[문제]

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

 

 

 

[코드]

def solution(n):
    big = n
    while True:
        big += 1
        if str(bin(n))[2:].count('1') == str(bin(big))[2:].count('1'):
            break

    return big
728x90