출발지점부터 distance만큼 떨어진 곳에 도착지점이 있습니다. 그리고 그사이에는 바위들이 놓여있습니다. 바위 중 몇 개를 제거하려고 합니다.
예를 들어, 도착지점이 25만큼 떨어져 있고, 바위가 [2, 14, 11, 21, 17] 지점에 놓여있을 때 바위 2개를 제거하면 출발지점, 도착지점, 바위 간의 거리가 아래와 같습니다.
제거한 바위의 위치 | 각 바위 사이의 거리 | 거리의 최솟값 |
---|---|---|
[21, 17] | [2, 9, 3, 11] | 2 |
[2, 21] | [11, 3, 3, 8] | 3 |
[2, 11] | [14, 3, 4, 4] | 3 |
[11, 21] | [2, 12, 3, 8] | 2 |
[2, 14] | [11, 6, 4, 4] | 4 |
위에서 구한 거리의 최솟값 중에 가장 큰 값은 4입니다.
출발지점부터 도착지점까지의 거리 distance, 바위들이 있는 위치를 담은 배열 rocks, 제거할 바위의 수 n이 매개변수로 주어질 때, 바위를 n개 제거한 뒤 각 지점 사이의 거리의 최솟값 중에 가장 큰 값을 return 하도록 solution 함수를 작성해주세요.
바위의 개수
이하입니다.distance | rocks | n | return |
---|---|---|---|
25 | [2, 14, 11, 21, 17] | 2 | 4 |
문제에 나온 예와 같습니다.
※ 공지 - 2020년 2월 17일 테스트케이스가 추가되었습니다.
문제를 풀 때 먼저 생각한 방법은 n개를 고르고 완전 탐색하는 방법이었다 하지만 그 방법을 실행했을 때
최악의 경우일 때 시간 초과
를 발생시킨다는 것을 알고 다른 방법을 찾았다.
다른 방법은 문제를 다르게 생각해보는 것이었다.
특정 값에서 n개를 지울 수 있는지?
를 생각하는 방법이었다.
특정 값(돌의 길의 최솟값)보다 작은 사이에 값을 가질 때마다 제거할 수 있는 돌로 remove++
해준다
모든 거리를 비교하고 나서, 제거할 수 있는 돌이 n개보다 큰 경우, 작은 경우에 따라서 범위를 재 조정해간다
import java.util.Arrays;
class Solution {
public int solution(int distance, int[] rocks, int n) {
int answer = 0;
Arrays.sort(rocks);
int start = 0;
int end = distance;
while (start <= end) {
int mid = (start + end) / 2;
int remove = 0;
int prev = 0;
for (int i = 0; i < rocks.length; i++) {
if (rocks[i] - prev < mid) {
remove++;
} else {
prev = rocks[i];
}
}
if (distance - prev < mid) remove++;
if (remove <= n) {
answer = Math.max(answer, mid);
start = mid + 1;
} else {
end = mid - 1;
}
}
return answer;
}
}