[프로그래머스] '징검다리' 알고리즘 풀이 - Python

Contents

1. 문제 설명
   
   [제한사항]
   
   [입출력 예]
2. 알고리즘 분석 
   
   [나의 풀이]
   
   [Most 1 의 풀이]

---

 

문제 설명

 

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

 

알고리즘 분석

 

• 다른 사람의 풀이

import math

def solution(distance, rocks, n):
    rocks.sort()
    rocks.append(distance)
    left, right, N = 0, distance, len(rocks)

    answer= 0
    while left <= right:
        prev = 0  # 이전 돌
        mins = math.inf  # 돌 거리의 최솟값
        removed_rocks = 0  # 제거한 돌의 개수

        mid = (left + right) // 2  # 바위 사이의 최소 거리
        
        for i in range(N):
            if rocks[i] - prev < mid: # 두 돌의 거리가 mid보다 작다면 
                removed_rocks += 1  # 돌을 삭제한 것으로 가정 
            else:
                mins = min(mins, rocks[i] - prev)
                prev = rocks[i]

        if removed_rocks > n:  # 삭제한 돌이 n보다 많으면 
            right = mid - 1
        else:  # 삭제한 돌이 n보다 적거나 같으면 
            answer = mins
            left = mid + 1
    return answer

 

이분 탐색이 익숙하지 않고 무엇을 기준으로 삼아야할지도 감이 오지 않아서 

다른 블로그를 참고했습니다. 

 

 

 

[Python] 프로그래머스. 징검다리 (Level 4)

 

 

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