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

https://school.programmers.co.kr/learn/courses/30/lessons/12911

프로그래머스

 

 

문제 설명

자연수 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 = 78
# result = 83

def dec_to_bin(s,bin=""):
    if s==0: return bin
    return dec_to_bin(s//2,bin=str(s%2)+bin)

n_bin = dec_to_bin(n)

n_bin = n_bin[::-1]
try:
    idx = n_bin.index("10")
    n_ = n_bin.replace("10","01",1)
    n_ = "".join(sorted(n_bin[:idx],reverse=True))+n_[idx:]
    answer = int(n_[::-1],2)
except:
    n_bin = n_bin[::-1]
    answer = int("10"+ "".join(sorted(n_bin[1:])),2)
    
print(answer)

먼저 십진법 > 이진법으로 만드는 함수부터 만들고 

 

주어진 수를 이진법으로 변환 > 순서 반전 > 첫번째 나오는 index를 찾고 10을 01로 replace했다.

find를 쓰지 않고 index를 쓴 것은 find는 없으면 -1을 리턴하지만 index는 error가 뜨기 때문에

try - except로 처리하기 편하기때문. if - else보다 이 방식을 선호한다.

 

이해만 하면 쉬운 수학문제

 

def solution(n):
    def dec_to_bin(s,bin=""):
        if s==0: return bin
        return dec_to_bin(s//2,bin=str(s%2)+bin)
    n_bin = dec_to_bin(n)
    n_bin = n_bin[::-1]
    try:
        idx = n_bin.index("10")
        n_ = n_bin.replace("10","01",1)
        n_ = "".join(sorted(n_bin[:idx],reverse=True))+n_[idx:]
        answer = int(n_[::-1],2)
    except:
        n_bin = n_bin[::-1]
        answer = int("10"+ "".join(sorted(n_bin[1:])),2)
    return answer