[고득점kit] 이분탐색 #2. 징검 다리

( 문제 설명 )

출발지점부터 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 이상 바위의 개수 이하입니다.

입출력 예distancerocksnreturn

25

[2, 14, 11, 21, 17]

2

4

입출력 예 설명

문제에 나온 예와 같습니다.

 

---

( 문제 해설 )

 

---

[ 코드 ]

 

# 2. 징검 다리
def solution(distance, rocks, n):
    answer = 0
    left = 0
    right = distance
    rocks.sort()
    #####
    while left <= right:
        # 미드값 이하는 전부 제거해준다
        mid = (left + right) // 2
        removeRock = 0
        prevRock = 0
        # min_a = 1000000001
        ##
        for rock in rocks:
            # 만약 현재 돌과, prev돌의 거리가 mid보다 작으면, 현재 돌을 제거한다
            if rock - prevRock < mid:
                removeRock += 1
            # 그게 아니라면, 현재돌 prev로하고 다시 체크
            else:
                # min_a = min(min_a, rock - prev_rock)
                prevRock = rock

        # 삭제한 돌이 n보다 많다면 mid를 줄여서 다시 확인
        if removeRock > n:
            right = mid - 1
        else:
            answer = mid
            left = mid + 1
        # else:
        #     answer = mid
        #     break
    return answer

---

후기

 

설명이 난잡하지만 주요 포인트는 그림 1에서

밑줄 친 부분입니다.

이 부분으로 맥락을 잡고 코드로 디테일을 구현하면 풀 수 있습니다