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

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

코딩테스트 연습 - 2개 이하로 다른 비트

 

쉽다. 짝수일 때 이진수는 0 으로 끝나기 때문에 + 1 을 해주면 된다. 홀수일 때는 2가지 경우의 수를 두어 잘 풀면 된다.

파이썬에 내장된 함수를 통하여 쉽게 진수 변환을 할 수 있다.  

 

def solution(numbers):
    answer = []

    for i in numbers:
        
        if i % 2 == 0:
            answer.append(i + 1)
        else:
            b = bin(i)[2:]
            b = list(b)
            
            for j in range(len(b)-1, -1, -1):
                if b[j] == '0':
                    b[j] = '1'
                    b[j+1] = '0'
                    b[0] = '0b' + b[0]
                    break
                
                if j == 0 and b[j] == '1':
                    b[0] = '0b10'
                    break
                    
            answer.append(int(''.join(b), 2))
                
    return answer

print(solution([2, 7])) # [3, 11]