[Programmers] 이분탐색 > 징검다리 with python

문제 설명

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

풀이

# 이분탐색의 기준 : 돌과 돌 사이의 거리...!

def solution(distance, rocks, n):
    answer = 0
    rocks.sort()

    left, right = 0, distance

    while left<=right:
        mid = (left+right)//2     # mid 거리값과 현재 위치와 바위 사이의 거리를 비교해 그 바위를 제거할지 말지 판단
        curr = 0                  # 현재 내 위치
        min_dist = 1000000001     # 한 시행에서의 최소 거리값
        rock_cnt = 0              # 제거된 바위 개수

        for rock in rocks:
            if rock-curr < mid:
                rock_cnt+=1       # mid 보다 작으면 그 바위 제거
            else:
                min_dist = min(min_dist, rock-curr)
                curr = rock       # mid 보다 크면 최소 거리값 넣어주고 그 바위 위치로 이동
        if rock_cnt > n:          # n보다 제거된 바위 개수가 더 많으면 right 이동
            right = mid-1
        else:
            answer = min_dist
            left = mid+1

    return answer