프로그래머스 [Lv. 2] 2개 이하로 다른 비트 {언어 : Python}

문제 링크

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

프로그래머스

정답 코드

def solution(numbers):
    answer = []
    for num in numbers:
        if num % 2:
            newNum = ["0"] + list(bin(num)[2:])
            lenN = len(newNum)
            zeroIdx = 0
            # 가장 가까이 있는 0을 1로 바꾸고
            # 다시 바꾼 지점부터 가장 가까운 1을 0으로 바꿈
            for i in range(lenN-1, -1, -1):
                if newNum[i] == "0":
                    newNum[i] = "1"
                    zeroIdx = i
                    break
            for j in range(zeroIdx+1, lenN):
                if newNum[j] == "1":
                    newNum[j] = "0"
                    break
            answer.append(int("".join(newNum), 2))
        else:
            answer.append(num+1)
    return answer

풀이 방법

1. 2의 배수인 경우 오른쪽 맨 끝 비트가 0이므로 1로만 바꿔줘도 현재 수보다 크지만 가장 작은 수가 된다.
2. 하지만 홀수인 경우 오른쪽 맨 끝 부터 0을 찾아본다.
   1. 0을 찾으면 1로 바꾸고
   2. 바꾼 지점에서 다시 오른쪽으로 이동하면서 1을 찾고 0으로 바꾼다.
3. 현재 만든 값을 int(숫자, n진수)를 사용해서 10진수로 만든다.
4. answer에 넣는다.

느낀점

• xor과 shift에 대해서 공부하면 좋을 것 같다.