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

문제 정보

제목 : 2개 이하로 다른 비트

난이도 : Lv.2

사용 언어 : Python

문제 링크

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

 

문제 설명

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

 

입출력 예 설명

입출력 예 #1

• 문제 예시와 같습니다.

---

나의 풀이

numbers를 순차적으로 접근하는 for루프를 생성하고, s에 num을 2진수로 변환한 값을 저장합니다. numbers의 원소를 2로 나눈 나머지가 0이라면 (짝수라면) 2진수를 int형으로 변환 후 1을 더해줍니다. 0이 아니라면 '0' + s를 s에 저장 후 s의 길이만큼 역순으로 반복하는 for 루프를 생성하여 s[i]가 0이라면 i와 i+1의 값을 바꾸어주고 break합니다. answer에 s를 10진수로 변환하여 저장합니다.

효율성은 떨어지는 편이지만 해결하는데 의미를 둔 문제였습니다.

 

코드

def solution(numbers):
    answer = []
    
    for num in numbers :
        s = bin(num)[2:]
        if num % 2 == 0 :
            s = int(s)+1
            
        else :
            s = '0' + s
            for i in range(len(s)-1,-1,-1) :
                if ( s[i] == '0') :
                    tmp = list(s)
                    tmp[i],tmp[i+1] = '1','0'
                    s = "".join(tmp)
                    break
        answer.append( int(str(s),2) )            
    return answer