[프로그래머스] (이분탐색) 징검다리

 

programmers.co.kr/learn/courses/30/lessons/43236

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

 

 

 

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

 

입출력 예

distancerocksnreturn

25

[2, 14, 11, 21, 17]

2

4

 

입출력 예 설명

문제에 나온 예와 같습니다.

 

 

2. 풀이 

 

1) 이중 분할이 아닌 완전 탐색으로 풀면 // --> 런타임 에러 

from itertools import combinations
from collections import deque


def solution(distance, rocks, n):

    combi_list = combinations(rocks,n)
    # print(combi_list)

    if 0 not in rocks:
        rocks.append(0)
    if distance not in rocks:
        rocks.append(distance)

    rocks = sorted(rocks)
    now_rock = rocks.copy()

    max_min_gap = -10000000000000
    for one_combi in combi_list:
        for one_n in one_combi:
            now_rock.remove(one_n)

        # print(now_rock)
        gap_list = []
        min_gap = 10000000000000000
        for idx in range(len(now_rock)-1):
            gap = now_rock[idx+1] - now_rock[idx]
            gap_list.append(gap)
            min_gap = min(min_gap, gap)

        now_rock = rocks.copy()
        max_min_gap = max(max_min_gap,min_gap)

    return max_min_gap

 

 

==> 무조건 이중 분할로 풀어야함.. 어떻게 풀어야하나 ? 

 

돌의 최소 간격을 탐색하는 이중분할임 

초기 설정 : 시작점 (left) 1 / 끝점 (right) distance  --> 이중 탐색 시작 

 

def solution(distance, rocks, n):
    answer = 0
    left = 1
    right = distance
    rocks.sort()
    while left <= right:
        mid = (left + right) // 2
        pre_rock = 0
        num_of_rock = 0
        m_min = 1000000001
        for rock in rocks:
            if rock - pre_rock < mid:
                num_of_rock += 1
            else:
                m_min = min(m_min, rock - pre_rock)
                pre_rock = rock

        if num_of_rock > n:
            right = mid - 1
        else:
            answer = m_min
            left = mid + 1
            
    return answer