99클럽 코테스터디 14일차 TIL 징검다리

문제 : 징검다리

프로그래머스 Level 4 이분탐색
문제 링크

문제 설명

출발지점부터 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 의 최대 길이가 약 10억 정도 되는데, 평범하게 완전탐색을 하거나 하면 시간초과가 남을 알 수 있다.

 

주어진 바위들 중에서 주어진 개수 만큼 바위를 제거하고, 각 바위의 거리의 최소값 중 최대값을 구하는 문제이기 때문에, 거리를 중점적으로 봐줬다.

 

출발지, 바위, 도착지 까지의 각각의 거리가 설정해준 값보다 작으면 해당 위치의 바위를 제거해주는 식으로 풀이했다.

코드

#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());

    int left = 0, right = distance;
    while (left <= right) {
        int mid = left + (right - left) / 2;

        int prev = 0, cnt = 0;
        for (int i=0; i<rocks.size(); i++) {
            if (rocks[i] - prev < mid)
                cnt++;
            else 
                prev = rocks[i];
        }

        if (distance - prev < mid) cnt++;

        if (cnt <= n) {
            left = mid+1;
            answer = mid;
        } else {
            right = mid - 1;
        }
    }

    return answer;
}

회고

오늘도 어제에 이어서 이분탐색 문제였다.

만약 어제 이 문제가 나오고 이분탐색이라는 것을 확인하지 못했다면 못풀었을 것 같은 문제였다.

 

지금까지 코테준비를 하면서 거의 시뮬레이션 계열의 bfs 탐색만 했더니 이런 Parametric Search 류의 문제는 거의 감을 못잡는 것 같다.

처음에 문제에 접근했을 때, 마지막 바위에서 도착지까지의 거리는 생각하지 못했었는데, 프로그래머스에 AI 분석이 알려줬다.

값을 갱신하는 과정에서 좀 애를 먹었었는데, 혹시나 하는 마음에 gpt 한테 이분탐색할 때, 어떻게 값을 갱신하는게 좋을지 물어봤더니 답을 줬다.

최대값을 찾는 경우: 조건을 만족할 때 answer를 갱신하고 left를 증가시킵니다.
최소값을 찾는 경우: 조건을 만족할 때 answer를 갱신하고 right를 감소시킵니다.

이게 항상 정답일 수는 없겠지만, 어제 문제와 오늘 문제를 생각해보면 어느정도 맞는 것 같아서 앞으로는 적용시켜볼 생각이다.