[프로그래머스] Lv2. 2개이하로 다른 비트

코드

def solution(numbers):
    arr = []
    answer =  []
    for num in numbers:
        arr.append('0'+bin(num)[2:])
    
    for i in range(len(numbers)):
        if numbers[i]%2 == 0: #짝수이면 가장 오른쪽 비트 1로
            answer.append(int(''.join(arr[i][:-1]+'1'),2))
        else:
            idx = arr[i].rfind('0') #홀수이면 0 찾아서 1로 바꾸고 그 오른쪽 0으로
            answer.append(int(''.join(arr[i][:idx]+'1'+'0'+arr[i][idx+2:]),2))
    return answer

 풀이

• numbers의 원소별로 해당 수보다 큰 값을 이진수로 바꿔가면서 비교하려고 했는데, 시간 초과가 발생해서 풀이를 찾아보니, 생각보다 간단했다.
• 짝수의 경우는 2^0의 비트가 0이므로, 1로 바꿔주면 해당 숫자보다 1 크면서 비트가 1개 다르므로 비트가 1~2개 다른 가장 작은 수이다.
• 홀수의 경우는 예를들어 7인 경우 0111에서 1111로 바꾸면 15, 1개가 다르면서 가장 작은 숫자일 수 있으나 2개까진 허용이므로 1111에서 오른쪽 비트를 하나씩 줄여보면
  • 1011: 11
  • 1101: 13
  • 1110: 14

로 나타낼 수 있으므로,1개~2개를 바꾼 비트중에 가장 작은 수를 찾으려면 0을 1로 바꾼 바로 오른쪽 비트를 바꾸면 된다.