프로그래머스 - 2개 이하로 다른 비트 Swift (비트)

2개 이하로 다른 비트

문제 설명

양의 정수 x에 대한 함수 f(x)를 다음과 같이 정의합니다.

• x보다 크고 x와 비트가 1~2개 다른 수들 중에서 제일 작은 수

예를 들어,

• f(2) = 3 입니다. 다음 표와 같이 2보다 큰 수들 중에서 비트가 다른 지점이 2개 이하이면서 제일 작은 수가 3이기 때문입니다.

수비트다른 비트의 개수

2	000...0010
3	000...0011	1

• f(7) = 11 입니다. 다음 표와 같이 7보다 큰 수들 중에서 비트가 다른 지점이 2개 이하이면서 제일 작은 수가 11이기 때문입니다.

수비트다른 비트의 개수

7	000...0111
8	000...1000	4
9	000...1001	3
10	000...1010	3
11	000...1011	2

정수들이 담긴 배열 numbers가 매개변수로 주어집니다. numbers의 모든 수들에 대하여 각 수의 f 값을 배열에 차례대로 담아 return 하도록 solution 함수를 완성해주세요.

---

제한사항

• 1 ≤ numbers의 길이 ≤ 100,000
• 0 ≤ numbers의 모든 수 ≤ 1015

---

입출력 예numbersresult

[2,7]

[3,11]

---

풀이 및 소스코드

생각보다 간단한 문제. 

짝수의 2진수는 모두 0으로 끝난다.

홀수의 2진수는 모두 1로 끝난다.

큰수 중 수와 최대한으로 가까워야 한다.

 

짝수를 생각해보면

2진수 10진수

0000(0) -> 0001(1)

0010(2) -> 0011(3)

0110(6) -> 0111 (7)

1000(8) -> 1001(9)

1110(14) -> 1111(15)

1100(12) -> 1101(13)

마지막 0만 1로 바꿔준다? -> "1을 더한다"라는 결과가 나온다.

 

홀수를 생각해보면

2진수 10진수

1) 0111(7) -> 1011(11)

2) 1001(9) -> 1010(10)

3) 1011(11) -> 1101(13)

4) 10111(23) -> 11011(27)

5) 0001(1) ->  0010(2)

 

마지막으로 등장하는 0을 1로 바꾸고, 다음 숫자를 0으로 바꾼다.

변환 전: 0111    1001

변환 후: 1011    1010

 

만약 7일경우 111로 변환이 되는데, 그냥 첫 숫자가 1일경우 0을 추가해주면 수의 변화는 없으므로 추가해주면 된다.

import Foundation

func solution(_ numbers:[Int64]) -> [Int64] {
    var result = [Int64]()
    
    for num in numbers{
        if num % 2 == 0{
            result.append(num + 1)
        }else{
            var binary = Array(String(num, radix: 2))
            if binary[0] == Character("1"){
                binary.insert(Character("0"), at: 0)
            }
            for idx in (0..<binary.count).reversed() {
                if binary[idx] == Character("0"){
                    binary[idx] = Character("1")
                    binary[idx+1] = Character("0")
                    break
                }
            }
            result.append(Int64(String(binary), radix: 2)!)
        }
    }
    
    return result
}