[알고리즘][X] 징검다리

문제 설명
출발지점부터 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
입출력 예 설명
문제에 나온 예와 같습니다.

출처

※ 공지 - 2020년 2월 17일 테스트케이스가 추가되었습니다.
※ 공지 - 2023년 5월 15일 테스트케이스가 추가되었습니다. 기존에 제출한 코드가 통과하지 못할 수도 있습니다.

참고해서 푼 풀이

- inf는 float('inf')로도 가능하다 이건 항상 다른 숫자보다 무조건 크다고 나온다

- 이분탐색은 기준을 어떻게 정하는지가 중요하다

- answer를 언제 할당하는지도 중요한데 여기서는 del_stones <= n일 때이다. 왜냐하면 문제에서 구하고자 하는 값은 최대이기 때문이다. 만약 반대로 된다면 del_stones == n일 때 오른쪽에서 왼쪽으로 즉 최솟값을 찾는 형태로 갈 것이기 때문이다.

def solution(distance, rocks, n):
    # 최솟값 중의 최댓값을 구하는 문제이니 그리디, 탐색, DP
    # 바위가 50,000개일 때 50000Cn이 되버리면 상당한 경우의 수가 발생 <= 전부 탐색 X = 그리디
    # 항상 최선의 선택을 할 수 있는가? 지금 한 선택이 최선이라 할 수 있는가? 아니다 자신없다
    # 그러면 남은 건 이분탐색이다.
    # 어떻게 풀어야 할지 감이 안온다.
    
    answer = 0
    left, right = 0, distance
    stones = rocks
    stones.sort()
    stones.append(distance)
    
    while left <= right:
        mid = (left + right) // 2
        del_stones = 0
        std_stone = 0
        min_distance = float('inf')
        for stone in stones:
            if stone - std_stone < mid:
                del_stones += 1
            else:
                min_distance = min(min_distance, stone - std_stone)
                std_stone = stone
            
            if del_stones > n:
                break
        
        if del_stones > n:
            right = mid - 1
        else:
            answer = min_distance
            left = mid + 1
            
    return answer​

참고해서 푼 풀이

 

Reference

---

https://soohyun6879.tistory.com/180

[프로그래머스/Python] 징검다리 - Level4