[프로그래머스] 징검다리(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 이상 바위의 개수 이하입니다.

전체코드(Python)

def remove_rock(arr,mid,distance):
    before = 0
    box = []
    for i in range(len(arr)):
        # 돌간의 거리가 mid보다 작다면 제거할 box에 담는다.
        if arr[i]-before < mid:
            box.append(arr[i])
        # 돌간의 거리가 mid보다 클 경우 이전 돌을 현재의 돌로 교체한다.
        else:
            before = arr[i]
    # 마지막 돌과 길이와의 간격이 mid보다 작을경우 마지막 돌도 제거할 box에 담는다.
    if distance-arr[-1] < mid:
        box.append(arr[-1])
    
    return len(box)
def solution(distance, rocks, n):
    answer = 0
    # 이분탐색을 통해 돌들간의 적정거리를 구해야하므로 오름차순으로 정렬한다.
    rocks.sort()
    box = 0
    start = 0
    end = distance
    # 이분 탐색을 통해 적절한 돌의 거리를 구한다.
    while start<=end:
        mid = (start+end)//2
        box = remove_rock(rocks,mid,distance)
        # box의 크기(제가한 돌)이 n보다 작거나 같을경우 start의 범위를 mid보다 하나 높인다.
        if box<=n :
            answer = answer if mid < answer else mid
            start = mid+1
        else:
            end = mid-1
    return answer

[참조]

https://programmers.co.kr/learn/courses/30/lessons/43236#

https://gom20.tistory.com/204

[프로그래머스] 징검다리 (Java)