Algorithm & Data Structures/Programers

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

Geisha 2024. 7. 23. 19:18

 

코드 흐름

  • for 문을 통해서 numbers 에서 하나씩 long 타입 숫자를 가져온다.
  • 그 숫자가 짝수이면 1 을 더한것이 값이되는 규칙이 있으므로 1을 더해서 answer 배열에 집어 넣는다.
  • 그 숫자가 홀수라면 처음 0이 나오는 구간에서 "10"으로 교체해준다.
  • 만약 1111과 같이 0이 없어 idx가 -1이된다면 01111로 생각하여 10111로 바꾼다.
  • Long.parseLong(a,b) 함수를 이용하여 2진수로 바꾸고 정답을 answer 배열에 넣는다.
import java.util.*;

class Solution {
    public long[] solution(long[] numbers) {
        long[] answer = new long[numbers.length];
        for(int i = 0 ; i < numbers.length; i++){
            String str = Long.toBinaryString(numbers[i]);
            if(numbers[i]%2==0)
                answer[i] = numbers[i]+1;
            else {
                int idx = str.lastIndexOf("0");
                
                if (idx == -1) {
                    String tmp = "10" + str.substring(1, str.length());
                    answer[i] = Long.parseLong(tmp, 2);
                } else {
                    String tmp = str.substring(0, idx) + "10" + str.substring(idx + 2, str.length());
                    answer[i] = Long.parseLong(tmp, 2);
                }
            }
        }
        return answer;
    }
}