[알고리즘] 프로그래머스 연습문제 - 다음 큰 숫자

프로그래머스 연습문제 - 다음 큰 숫자

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

비교적 간단한게 풀었던 문제이지만, 구글링이 불가능한 코딩 테스트라면 조금 버벅일 수 있을만한 문제였다. 이 문제에서는 bin() 함수를 사용해야 하는데 사용해 본 적이 없어서 조금 난감했을 것 같다.

해당 문제는 자연수 n이 주어졌을 때, n 의 다음 큰 숫자를 구하는 문제이고 조건은 아래와 같다.

자연수 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이 들어온다면, 79부터 검사를 시작하여 1의 개수가 맞는 수를 리턴하면 된다. 이 때 bin() 을 사용하여 1의 개수를 판단한다.

• bin() 함수 예제

  >>> n = 78
  >>> n_to_bin = bin(n)
  >>> print(n_to_bin)
  0b1001110
  
  # 결과는 문자열로 리턴된다.

• 내가 작성한 코드

  def solution(n):
      answer = 0
      n_to_bin_list = list(bin(n)[2:])
      n_one_total = n_to_bin_list.count('1')
      next_n = n + 1
  
      while True:
          next_n_to_bin_list = list(bin(next_n)[2:])
          next_n_one_total = next_n_to_bin_list.count('1')
          if n_one_total == next_n_one_total:  
              answer = int("".join(str(i) for i in next_n_to_bin_list), 2)
              break
          next_n += 1
      return answer