[프로그래머스] 징검다리

문제 설명

출발지점부터 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는 1 이상 1,000,000,000 이하입니다.
• 바위는 1개 이상 50,000개 이하가 있습니다.
• n 은 1 이상 바위의 개수 이하입니다.

입출력 예

distance	rocks	n	return
25	[2, 14, 11, 21, 17]	2	4

풀이)

이분탐색 문제이다.

distance가 1 ~ 1,000,000,000으로 되어있어서 바위중 n개를 고르는 완전탐색시 시간초과 난다.

이분탐색 값의 기준은 바위 사이의 거리이다. (start = 1, end = distance, mid = (start + end) / 2)

두 바위 사이의 거리가 mid보다 작을경우, 뒤쪽에 있는 바위를 없애주었다.

그리고, 없애준 바위의 개수를 세어서 n보다 작으면 더 제거해줘도 된다는 뜻이니깐 거리기준을 높이고

n보다 크면 제거해줘야 하는 바위를 줄여야한다는 것이니깐 거리기준을 낮춰줬다.

코드)

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int solution(int distance, vector<int> rocks, int n) {
    int answer = 0;
    sort(rocks.begin(),rocks.end());
    rocks.push_back(distance); //바위 사이의 거리 계산 위해서 삽입
    int start =1;
    int end = distance;
    int mid, temp, val;
    while(end>=start){
        temp = 0;
        mid = (start + end) / 2; // 바위 사이 최소 거리
        int j =0, val = 999999999;
        for(int i=0;i<rocks.size();i++){
            if(rocks[i] - j < mid){
                temp++;
                if(temp > n) break;
            }
            else{
                val = min(val,rocks[i] - j);
                j = rocks[i];
            }
        }
        if(temp > n){ //제거한 돌 개수가 많다 -> 제거 범위 줄여야함
            end = mid - 1;
        }
        else{
            answer = max(answer,mid);
            start = mid + 1;
        }
    }
    return answer;
}