[파이썬] 프로그래머스 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 = 25, rocks = [2, 14, 11, 21, 17] , n = 2

return 4

 

---

풀이

 

이분탐색 문제에서는 어떤 것을 이분탐색의 기준으로 삼을지 판단해야 한다.

이 문제에서는 돌과 돌 사이의 거리가 그 기준이다. 

여기서 돌과 돌 사이의 거리는 rocks[i] - prev 값이다.

min_inter값은 리턴되는 값을 구하기 위해 간격들의 최솟값을 구하기 위한 변수이다. math.inf 는 무한대 값으로, 최솟값을 구하기 위한 초기값을 넣어두는 것이다.

 

import math
def solution(distance,rocks, n):
    answer=0
    start,end=0,distance
    rocks=sorted(rocks)
    rocks.append(distance)
    rnum=len(rocks)
 
    while start<=end:
        remove=0
        prev=0
        min_inter=math.inf
        mid = (start+end)//2
 
        for i in range(rnum):
            if rocks[i]-prev < mid:  #rocks[i]-prev가 inter값
                remove+=1 #바위제거
                
            else : #바위 제거 안할경우 
                min_inter=min(min_inter,rocks[i]-prev) #inter값 갱신.
                prev=rocks[i] #현재 바위위치를 prev로
 
        #너무 많이 제거되었을 경우, 기준을 높여서 제거를 줄여야함
        if remove > n:
            end=mid-1
        #적게 제거되었을 경우, 기준을 낮춰서 제거를 늘려야함
        else:
            answer=min_inter
            start=mid+1
            
    return answer

이 문제를 풀 때, 일단 어느 것을 이분탐색의 기준으로 잡을지 판단하는 것조차 너무 어려웠고, 그 기준을 잡고 나서도, 코드를 써내려가는 것이 굉장히 어려웠다. 다른 해설들을 참고하고 나서야 그나마 이해가 되었던 문제였다. 초보자인 지금은 이런 문제를 척척 풀 수 없지만 이해하고 코드를 분석하는 데에 의의를 두어야겠다.