💻코딩테스트 문제풀이23

Chapter12. 구현

[문제53] 2개 이하로 다른 비트 - Level2

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

[코드작성]

1. f(x) 함수 만들기

def f(x):
	if x % 2 == 0: return x + 1
    
    x = f'0{bin(x)[:2]}'
    x = f"{x[:x.rindex('0')]}10{x[x.rindex('0') + 2:]}"
    return int(x, 2)

2. 단어를 받아 트라이에 데이터 채우기

def solution(numbers):
	return [f(number) for number in numbers]

[전체코드]

def f(x):
	if x % 2 == 0: return x + 1
    
    x = f'0{bin(x)[:2]}'
    x = f"{x[:x.rindex('0')]}10{x[x.rindex('0') + 2:]}"
    return int(x, 2)
    
def solution(numbers):
	return [f(number) for number in numbers]