본문 바로가기

프로그래머스/Java

2개 이하로 다른 비트

문제


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

입출력 예

[2,7] [3,11]

답안


class Solution {
    public long[] solution(long[] numbers) {
        long[] answer = new long[numbers.length];
        
        StringBuilder sb = new StringBuilder();
        
        for(int index=0; index<numbers.length; index++) {
        	if(numbers[index]%2 ==0) {
        		answer[index] = numbers[index] + 1;
        	}else {
                sb.setLength(0);
        		String binary = Long.toBinaryString(numbers[index]);
        		
        		int first = binary.lastIndexOf("0");
        		
        		if(binary.contains("0")) {
        			sb.append(binary.substring(0, first));
        			sb.append("10");
        			sb.append(binary.substring(first+2, binary.length()));
        		}else {
        			sb.append("10");
        			sb.append(binary.substring(1));
        		}
        		
        		answer[index] = Long.parseLong(sb.toString(), 2);
        	}
        }
        
        return answer;
    }
}

'프로그래머스 > Java' 카테고리의 다른 글

숫자 블록  (0) 2022.08.02
줄 서는 방법  (0) 2022.08.02
괄호 회전하기  (0) 2022.04.05
이진 변환 반복하기  (0) 2022.04.04
쿼드압축 후 개수 세기  (0) 2022.04.04