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

입출력

istancerocksnreturn
25[2, 14, 11, 21, 17]24


📖내가 작성한 code

def solution(distance, rocks, n):
    answer = 0
    rocks.sort()
    rocks.append(distance)
    lt,rt=1,distance
    while lt<=rt:
        pre=0
        removed_rocks=0
        mid=(lt+rt)//2
        for rock in rocks:
            if rock-pre<mid:
                removed_rocks+=1
            else:
                pre=rock
        if removed_rocks>n:
            rt=mid-1
        else:
            answer=mid
            lt=mid+1
    return answer

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/43236

좋은 웹페이지 즐겨찾기