[프로그래머스/Level2] 2개 이하로 다른 비트 (Python)

 

문제 설명

양의 정수 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

입출력 예

numbers	result
[2,7]	[3,11]

 

---

 

문제 풀이

💡접근 방법

• 짝수일 경우, 마지막 비트가 0이기 때문에 1을 더하면 조건에 만족한다.
• 홀수일 경우, 가장 뒤쪽에 있는 0의 인덱스(idx)를 1로 바꿔주고 그 뒤 인덱스(idx+1)를 0으로 바꿔준다.

 

✏️ 문제 풀이

def solution(numbers):
    answer = []
    
    for number in numbers:
        if number % 2 == 0:
            answer.append(number + 1)
            continue
        
        bin_number = '0' + format(number, 'b')
        zero_idx = bin_number.rfind('0')
        bin_number = list(bin_number)
        bin_number[zero_idx] = '1'
        bin_number[zero_idx+1] = '0'
        
        bin_number = ''.join(bin_number)
        answer.append(int(bin_number, 2))
        
    return answer

 

---

🔗 문제 출처

프로그래머스