[Programmers 프로그래머스] 77885 2개 이하로 다른 비트 (PYTHON)

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

코딩테스트 연습 - 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

---

입출력 예

 

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

 

 

먼저 숫자를 홀수일때와 짝수일 때를 나눠서 계산을 합니다.

짝수일 때는 숫자를 2진법으로 만들고 마지막에 1을 추가해 줍니다.

홀수일 때는 가장앞에 0을 추가를 해주고, 해당하는 0을 1로 바꾸고, 그 다음 숫자는 1로 바꿔줍니다.

그렇게 만들어진 숫자를 10진법으로 바꿔서 답을 출력해 줍니다.

def solution(numbers):
    answer = []
    for number in numbers:
        if number % 2 == 0:	#짝수 일때
            binary_num = list(bin(number)[2:])
            binary_num[-1] = "1"
        else:	#홀수 일때
            binary_num = bin(number)[2:]
            binary_num = "0" + binary_num
            one_idx = binary_num.rfind("0")
            binary_num = list(binary_num)
            binary_num[one_idx] = "1"
            binary_num[one_idx + 1] = "0"
        ans_num = int("".join(binary_num), 2)
        answer.append(ans_num)
        
    return answer