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

---

• 입/출력

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

---

• 코드

def solution(distance, rocks, n):
    answer = 0
    rocks.sort()
    rocks.append(distance)
    
    left = 0
    right = distance
    
    while left <= right:
        remove_rock = 0
        prev_rock = 0
        min_value = 1e+9+1
        
        mid = (left + right) // 2
        
        for rock in rocks:
            if rock - prev_rock <= mid:
                remove_rock += 1
            else:
                min_value = min(min_value, rock-prev_rock)
                prev_rock = rock
                
        if remove_rock > n:
            right = mid-1
        else:
            answer = max(answer, min_value)
            left = mid+1
    return answer

투포인터의 기준을 거리로 잡는것이다. 이부분을 생각해내는게 쉽지 않았다.

거리를 정하고 돌들과 비교하면서 천천히 제거를 해본다!

만약 제거한 개수가 더 많으면 거리를 줄여야하기 때문에 right를 mid-1로 바꾸어주고 진행한다.

반대라면 거리를 더 늘려야하기 때문에 left를 mid+1로 바꾸어주고 진행한다.

 

1) 돌들과 사이의 거리를 구할때는 최소값을 구해야하기때문에 min_value에 min 사용

2) 최소값 중에서는 최대값을 찾기때문에 answer는 max사용

---

• 출처

programmers.co.kr/learn/courses/30/lessons/43236?language=python3