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

2개 이하로 다른 비트 / Lv.2

---

문제  설명 )

양의 정수 x에 대한 함수 f(x)를 다음과 같이 정의합니다.

• x보다 크고 x와 비트가 1~2개 다른 수들 중에서 제일 작은 수

예를 들어,

• f(2) = 3 입니다. 다음 표와 같이 2보다 큰 수들 중에서 비트가 다른 지점이 2개 이하이면서 제일 작은 수가 3이기 때문입니다. 

• f(7) = 11 입니다. 다음 표와 같이 7보다 큰 수들 중에서 비트가 다른 지점이 2개 이하이면서 제일 작은 수가 11이기 때문입니다. 

정수들이 담긴 배열 numbers가 매개변수로 주어집니다. numbers의 모든 수들에 대하여 각 수의 f 값을 배열에 차례대로 담아 return 하도록 solution 함수를 완성해주세요.

 

 

 

제한 사항 )

• 1 ≤ numbers의 길이 ≤ 100,000
• 0 ≤ numbers의 모든 수 ≤ 10^15

 

 

 

입출력 예 )

 

 

 

입출력 예 설명 )

입출력 예 #1

• 문제 예시와 같습니다.

 

 

 

풀이)

 

#include <string>
#include <vector>
#include <bitset>
 
using namespace std;
 
vector<long long> solution(vector<long long> numbers) {
    vector<long long> answer;
    
    bitset<64> bit;
    
    for(int i = 0; i < numbers.size(); i++)
    {
        if(numbers[i] % 2 == 0)
        {
            answer.push_back(numbers[i] + 1);
        }
        else
        {
            bit = numbers[i];
            
            // numbers[i]는 홀수이기 때문에 bit[0]은 무조건 1이다.
            // 따라서 bit.set(i - 1, 0)이 인덱스 에러로 터질 이유는 없다.
            for(int i = 0; i < bit.size(); i++)
            {
                if(bit[i] == 0)
                {
                    bit.set(i, 1);
                    bit.set(i - 1, 0);
                    break;
                }
            }
            
            answer.push_back(bit.to_ulong());
        }
    }
    
    
    
    return answer;
}

---

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

프로그래머스