[Swift Algorithm] 다음 큰 숫자 Programmers(Lv.2)

문제 설명

자연수 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	result
78	83
15	23

 

내 문제 풀이

func solution(_ n:Int) -> Int {
    let binaryNum = String(n, radix: 2) // n을 이진수로 변경
    let count1 = binaryNum.map({$0}).filter({$0 == "1"}).count // n 이진수의 1 개수
    var result = n + 1 // n보다 1 큰 자연수
    
    while String(result, radix: 2).map({$0}).filter({$0 == "1"}).count != count1 {
        result += 1
    }

    return result
}

• 입력 값을 radix를 사용해 2진수로 변경해주었다.
• 2진수로 변경한 값은 String이기 때문에, map과 filter를 사용해 1의 개수를 찾았다.
• while을 사용해 n보다 큰 수의 2진수 1의 개수가 입력 값 2진수의 1 개수와 같을 때까지 숫자를 하나씩 키웠다.

 

💡 피드백

• 코드가 조금 복잡해서 가독성있게 바꾸면 좋을 것 같다.

 

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution(_ n:Int) -> Int {
    var nNext = n + 1
    let nBinary_1 = String(n, radix: 2).map{$0}.filter{$0 == "1"}.count
    
    while true {
        let nNextBinary_1 = String(nNext, radix: 2).map{$0}.filter{$0 == "1"}.count
        if nNextBinary_1 == nBinary_1 { return nNext }
        else { nNext += 1 }
    }
}

• 첫 번째 풀이와 로직은 같은데, while 조건을 true로 써버리고 조건을 안에 넣어주었다.
• 더 깔끔하고 괜찮은 것 같다.

 

---

 

문제

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