[Programmers] 징검다리(Binary Search)

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

코딩테스트 연습 - 징검다리

문제 설명

출발지점부터 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

입출력 예 설명

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

'''
프로그래머스 - 이분탐색 - 징검다리
    이분탐색을 생각하기 어려웠다. 이것또한 주어진 거리로 최대값을 구한 뒤 이 값을 바위간의 최소 거리로 설정
    시작돌부터 두 돌의 거리를 구해가면서 최소거리보다 작으면 명제에 어긋나므로 그 돌을 삭제한다고 생각하고 다음 돌로 넘어간다
    만약 최소거리보다 크면 그 돌이 이번턴의 최소거리 중 최소일 수 있으므로 비교 후 저장한다
'''
import math
def solution(distance, rocks, n):
    answer = 0
 
    rocks.sort()
 
 
    left, right = 0, distance  # 피봇범위 설정
    while left<=right:
        prev = 0 #이전 돌
        mmin = math.inf # 이번 피봇에서의 최소값
        mid = (left + right) // 2 # 피봇
        num_of_delete_rock = 0 # 이번 피봇에서 제거한 돌의 개수
        for rock in rocks:
            if rock - prev < mid: # 이번 피봇이 이번 턴에 설정한 최소거리이므로 그보다 작은 것은 명제에 어긋나니 제거
                num_of_delete_rock += 1
            else:
                mmin = min(mmin, rock - prev) # 그게 아니라면 이번 피봇보다 큰 값 중 최소 설정
                prev = rock
        if num_of_delete_rock > n: # 만약 이번 피봇에서 제거한 수가 n보다 크면 이번 피봇이 너무 큰것. 피봇을 줄인다
            right = mid-1
        else:
            answer = mmin # n보다 작거나 같다면 일단 mmin 저장
            left = mid + 1 # 피봇이 작다는 뜻이므로 피봇 키운다
    return answer