본문 바로가기

프로그래머스(LV2)

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

1. 문제설명

자연수 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

 

 

 


2. 문제풀이

def solution(n):
    
    #자연수 n을 이진수로 변환했을 때 1의 갯수
    cnt1 = bin(n)[2:].count("1")
    
    #자연수 n을 1씩 증가시킨다.
    while True:
        n += 1
        if cnt1 == bin(n)[2:].count("1"):
            break

    return n

접근방식

  • bin(n)[2:].count("1) 해석
  • 자연수 n을 bin( )을 통해 이진수로 변환한다. 
  • 그리고 count("1")을 이용해 n이라는 이진수에 "1"이 몇 개가 있는지 세고 cnt1에 저장한다.
  • cnt1 = 2진수 n의 1의 개수

 

  • while문에서는 자연수 n을 1씩 증가시키며 이진수로 변환했을 때 1의 개수가 cnt1와 같은지 확인한다.
  • cnt1와 값이 같으면 break, 다르면 n을 1씩 증가시키며 반복