Programmers - 징검다리[파이썬(python)]

문제 출처 : 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 이상 바위의 개수 이하입니다.

 

입출력 예

distance	rocks	n	return
25	[2, 14, 11, 21, 17]	2	4

 

---

문제 풀이

이 문제는 이분 탐색 문제로 분류되어 있다. 바위 간 최소 간격을 이분 탐색을 통해 찾으면 문제를 해결할 수 있다.

알고리즘은 다음과 같다.

1. 최소 간격은 1부터 최대 distance까지 될 수 있다.
2. mid를 최소 간격이라고 가정하고 mid를 만족할 수 있도록 바위를 없앤다.
3. 없앤 바위의 수를 확인한다.
   • 최소 간격 mid를 만족하기 위해 n보다 바위를 많이 없애야 한다면 최소 간격을 줄이기 위해 end값을 감소시킨다.
   • 최소 간격 mid를 만족하기 위해 n보다 바위를 적거나 같게 없앴다면 최소 간격을 늘리기 위해 start값을 증가시킨다.
4. 최대가 되는 최소 간격을 찾을때까지 2, 3 과정을 반복한다.

구현 코드

코드 링크 : 징검 다리

import copy
def solution(distance, rocks, n):
    rocks += [0, distance]
    rocks.sort()
    n_rocks = len(rocks)
    start = 1
    end = distance
    
    while start<=end:
        mid = int((start+end)/2)
        count = 0
        remain = 0
        dist = 0
        
        for i in range(1, n_rocks):
            dist = abs(rocks[remain]-rocks[i])
            if dist < mid:
                count += 1
            else:
                dist = 0
                remain = i
                
        if count > n:
            end = mid - 1
            
        else:
            start = mid + 1
            result = mid
            
    return result