프로그래머스 - 다음 큰 숫자

Posted on | In |

오늘의 문제

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

예를 들어서 78(1001110)의 다음 큰 숫자는 83(1010011)입니다.

자연수 n이 매개변수로 주어질 때, n의 다음 큰 숫자를 return 하는 solution 함수를 완성해주세요.

조건

  1. 조건 1. n의 다음 큰 숫자는 n보다 큰 자연수 입니다.
  2. 조건 2. n의 다음 큰 숫자와 n은 2진수로 변환했을 때 1의 갯수가 같습니다.
  3. 조건 3. n의 다음 큰 숫자는 조건 1, 2를 만족하는 수 중 가장 작은 수 입니다.
  4. n은 1,000,000 이하의 자연수 입니다.

풀이 방법

1
2
3
4
5
6
7
import collections
def solution(n):
    
    count_n = collections.Counter(str(bin(n)))
    for i in range(n+1, 1000001):
        if count_n['1'] == collections.Counter(str(bin(i)))['1']:
            return i

조건 4가 백만 이하의 자연수이기 때문에 for문으로 돌려주며, return i 로 값이 같으면 뒤에는 더 돌려 볼 필요도 없이 반환되게 만들어주기 때문에 효율성 부분에서 통과한다.

사담

일단 싹 다 출력하고 맨 앞에 걸 먼저 뽑아보려고 했다 ㅋㅋㅋ 바보였다.. ㅎ

1
2
3
4
5
import collections

def solution(number, k):
    count_n = collections.Counter(str(bin(n)))
    return [i for i in range(n+1, 1000001) if count_n['1'] == collections.Counter(str(bin(i)))['1']][0]