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

문제 설명

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

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

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

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

Code

class Solution {
    public long[] solution(long[] numbers) {
        long[] ret = new long[numbers.length];
        for(int i=0 ; i<numbers.length ; i++) {
            long tmp = numbers[i];
            int idx = 0;
            while(tmp>0) {
                if(tmp%2==0) break;
                tmp /= 2;
                idx++;
            }
            ret[i] = numbers[i]+(long)Math.pow(2,idx)-(idx>0?(long)Math.pow(2,idx-1):0);
        }
        return ret;
    }
}