[programmers] 2개 이하로 다른 비트 - Java

2개 이하로 다른 비트

문제 설명

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

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

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

제한사항

• 1 ≤ numbers의 길이 ≤ 100,000

• 0 ≤ numbers의 모든 수 ≤ 10
15

입출력 예

solution.java

class Solution {
    public long[] solution(long[] numbers) {
        // 결과를 저장할 배열
        long[] results = new long[numbers.length];
        
        // 입력 배열의 각 숫자 처리
        for (int i = 0; i < numbers.length; i++) {
            long x = numbers[i];
            
            if ((x & 1) == 0) {
                // x가 짝수인 경우, 최대 한 비트 차이가 있는 다음 숫자는 x + 1
                results[i] = x + 1;
            } else {
                // x가 홀수인 경우 x의 이진 표현에서 첫 번째 '0'을 찾는다.
                // 해당 0을 1로 바꾸고, 바꾼 1의 오른쪽 비트를 0으로 만들면
                // 비트가 2개 차이나는 가장 작은 수가 된다.
                long bitMask = 1;
                while ((x & bitMask) != 0) {
                    bitMask <<= 1;
                }
                results[i] = x + bitMask - (bitMask >> 1);
            }
        }
        
        return results;
    }
}

핵심 키워드

• & 연산자와 <<= 연산자를 통해 0 비트를 찾는다.

결론!