몸과 마음이 건전한 SW 개발자

프로그래머스 [Lv. 2] 다음 큰 숫자 {언어 : Python} 본문

알고리즘/풀었지만 다시 보기

프로그래머스 [Lv. 2] 다음 큰 숫자 {언어 : Python}

스위태니 2024. 11. 22. 14:25

문제 링크

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

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

정답 코드

def solution(n):
    answer = 0
    def binary(num):
        cnt = 0
        while num > 0:
            cnt += num % 2
            num //= 2
        return cnt
    one = binary(n)
    numDict = {}
    for num in range(n+1, 1000065):
        cnt = binary(num)
        if one == cnt:
            answer = num
            break
        
    return answer

풀이 방법

  1. 현재 n의 1개수를 측정한다.
  2. n+1 부터 탐색해서 1000064까지 탐색한다.
    1. 1000064인 이유는 n의 최대값 1000000보다 큰 수이자 1의 개수가 같은 수가 1000064이기 때문이다.
    2. 1의 개수가 같으면 바로 반복문 종료
  3. answer의 값을 출력하면 끝!

느낀점

  • 예전에는 이게 시간초과가 났었는데 이제는 바로 찾는 방법을 이해한 것 같다.