[Swift] lv.2 다음 큰 숫자 | Programmers

문제 설명

자연수 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 이하의 자연수 입니다.

1차 고민

음... 그러니까 이진수 기준으로 1의 갯수가 같은 수를 찾아오면 된다는거지?
이진수 1 갯수가 같은 거 나올 때까지 올리면 되겠구만...

1차 코드

import Foundation

func solution(_ n:Int) -> Int {
    var binary = String(n, radix: 2)
    let count = binary.filter{ $0 == "1" }.count
    var n = n + 1

    while true {
        binary = String(n, radix: 2)
        if binary.filter({ $0 == "1" }).count == count {
            break
        }
        n += 1
    }
    return n
}

1차 결과

아 신이시여...

2차 고민

어쩐지 쉽다했다. 또 뭘 빼먹었을까...

라고 하기전에 반드시 해야하는 작업. 프로그래머스에서 시간 초과 오류가 났다면 반드시 같은 코드로 돌려보고 의심하라.
엔터 추가하고 한 번 더 간다.

2차 코드

1차와 동일 (엔터로 한 줄만 추가)

2차 결과

이건 뭐임? 똑같이 또 엔터만 친다

3차 결과

참ㄴ 칵퉷 이녀석아