https://programmers.co.kr/learn/courses/30/lessons/77885
문제 설명
양의 정수 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
입출력 예
numbers | result |
---|---|
[2,7] | [3,11] |
문제 풀이
function solution(numbers) {
var answer = [];
for (let el of numbers) {
let binary = el.toString(2);
if(binary[binary.length-1] === '0') answer.push(el+1);
else {
let idx = binary.lastIndexOf("01")
let num = binary.substring(0,idx) + "10" + binary.substring(idx+2)
answer.push(Number.parseInt(num,2))
}
}
return answer;
}
number의 요소를 2진수로 바꿔준다. 그 뒤로 가장 마지막 수가 0 인 이진수는 number의 요소에 1을 더해줘서 값에 넣어준다.
그렇지 않을 경우 가장 마지막에 있는 문자열 "01"이 "10"으로 바뀌면 비트가 1~2개만 다른 수 중 number보다 큰 최소의 이진수 x가 나오게 되고 이진수 x를 10진수로 바꿔서 값에 넣어준 뒤 리턴하게 되면 정답이 나온다. 조건이 1~2개만 다른 비트인 숫자를 구하는 것인데 짝수의 경우 1의자리 이진수만 바꿔주면 되기에 1개만 다른 비트이고 홀 수의 경우 2개의 다른 비트를 가져야하고 이외에는 같아야하며 그중에서도 가장 작은 수를 구해야했기에 "01"을 "10"으로 바꿔주게 됐다.