출발지점부터 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 |
나의 답
# distance의 최대값이 10억까지 가능하므로 일반적인 탐색 방법으로는 시간 초과 날 듯
def solution(distance, rocks, n):
rocks.sort()
rocks.append(distance) # 도착점도 추가
left = 1
right = distance
answer = 0
while left <= right:
mid = (left + right) // 2
prev = 0
removed = 0
for rock in rocks:
if rock - prev < mid :
removed += 1 # 기준값 mid보다 간격이 작으면 바위 제거
else:
prev = rock # 간격 괜찮으면 다음으로 넘어가기
### removed 값 확정된 상태에서 ###
if removed > n: # 너무 많이 삭제한 경우 = 간격이 크다
right = mid - 1 # 간격을 줄인다
else: # n개 이하로 제거된 경우 = 더 큰 간격 해보기
answer = mid # 우선 값 저장
left = mid + 1 # 더 큰 간격 도전
return answer
- 이분 탐색을 해야한다는 건 사실 분류도 그렇게 되어 있기 때문에 떠올릴 수 있었지만 어떤 값을 기준으로 삼고 이분 탐색을 할 것이냐를 잘 떠올리는 건 쉽지 않은 것 같다.
- 여기서 키 포인트는 간격의 최솟값 x를 1부터 distance까지로 두고 x를 이분 탐색해야한다는 점이다.
- 이를 코드로 구현하였고 사실 키 값만 잘 잡으면 코드 구현 자체는 이분 탐색이 그나마 쉬운편에 속하는 것 같다.
- 문제를 보고 이분 탐색임을 떠올리고, 키 값을 바로 잡을 수 있게 더 많은 문제를 접해봐야겠다.
'공부기록 > TIL' 카테고리의 다른 글
[TIL] 250710 - 프로그래머스 다리를 지나는 트럭 (1) | 2025.07.10 |
---|---|
[TIL] 250622 - 프로그래머스 프로세스 (0) | 2025.06.22 |
[TIL] 250530 - 프로그래머스 네트워크 (0) | 2025.05.30 |
[TIL] 250529 - 프로그래머스 정수 삼각형 (0) | 2025.05.29 |
[TIL] 250528 - 프로그래머스 큰 수 만들기 (0) | 2025.05.28 |