[프로그래머스] Level2, 다음 큰 숫자 (Python)

문제

programmers.co.kr/learn/courses/30/lessons/12911

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

 

코드

def solution(n):
    answer = 0
    c = bin(n).count("1")
    for m in range(n+1, 1000001):
        if c == bin(m).count("1"):
            answer = m
            break
    return answer

python에 bin()을 이용해 이진수로 만들어주고 count()를 이용해 1의 갯수를 세어준다.

 

그리고 입력한 값보다 +1 한 값에서 부터 문제에 범위까지 for문을 돌아 1의 갯수가 같을 경우 바로 break문을 실행해 가장 작은 수를 구한다.