Programmers - 다음 큰 숫자(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 이하의 자연수 입니다.

입출력 예

nresult
7883
1523

입출력 예 설명

입출력 예#1
문제 예시와 같습니다.

입출력 예#2
15(1111)의 다음 큰 숫자는 23(10111)입니다.

✍ 코드

def solution(n):
    answer = 0


    # (1)
    cnt_one_bin_n_1 = bin(n).count('1')

    while 1:
        # (2)
        n = n + 1

        # (3)
        cnt_one_bin_n_2 = bin(n).count('1')

        # (4)
        if cnt_one_bin_n_2 == cnt_one_bin_n_1:
            # (5)
            answer = n

            # (6)
            break

    return answer

n = 15
print(solution(n)) # 결과 예 : 23 

(1) : n을 이진수로 변환한 후, 이진수 n의 1의 개수를 파악하귀 위한 변수 cnt_one_bin_n
(2) : n+1부터 차례대로 비교
(3) : n+1을 이진수로 변환한 후, 이진수 n의 1의 개수를 파악하귀 위한 변수 cnt_one_bin_n_2
(4) : cnt_one_bin_n_2와 cnt_one_bin_n_1이 동일할 경우 해당 if문 수행
(5) : answer에 n 대입
(6) : while문 탈출

📝 결과

😃 느낀점

  • 예전에 풀었던 프로그래머스 문제 중에 "[1]차 비밀지도" 문제와 매우 흡사한 것 같아서 쉽게 풀었던 것 같다.

👍 Irish의 모든 코드 보기

-> Irish Github

좋은 웹페이지 즐겨찾기