[Level 2] 다음 큰 숫자 (Python)

코딩테스트 연습 - 다음 큰 숫자

 

def solution(n):
    n_bin = bin(n)[2:]
    
    if len(set(n_bin)) == 1:
        return(int('10' + n_bin[1:], 2))
    
    one_cnt = n_bin.count('1')
    
    while(True):
        n += 1
        
        if bin(n)[2:].count('1') == one_cnt:
            return(n)

 

▷ n의 이진수가 모두 1인 경우와 아닌 경우로 나누어 주어야 한다.

 

▷ n의 이진수가 모두 1인 경우에는 set 함수를 이용하여 찾아준 뒤, 제일 앞에 '10'를 배치하고, 이진수의 나머지 부분을 붙여준다. 마지막으로 다시 int 함수를 이용하여 10진법으로 바꾸어 준다.

 

▷ n의 이진수가 모두 0이 아닌 경우는 while 함수를 통해 n에 계속 1씩 더하여 n의 이진수의 1의 개수가 같아지는 것을 찾아주면 된다.

 

▶ set 함수에 문자열이 들어오면 문자열의 문자에 대한 집합을 출력한다.

 

▶ count 함수는 문자열에 포함된 인자의 개수를 출력한다.