15-janghw0126

[janghw0126]

🔗 문제 링크

프로그래머스 | 징검다리

✔️ 소요된 시간

1시간 반

✨ 수도 코드

👉 문제 코드 바로가기

이 문제는 바위를 n개 제거하고 난 후에 각 지점 사이의 최솟값들 중에 최댓값을 찾아나가는 과정이므로 탐색을 이용해서 문제를 해결해야겠다고 생각했습니다.
하지만 문제를 보면, 순차탐색으로는 효율성을 만족하기 어렵기 때문에 이진탐색으로 다리를 건널 수 있는 최대 인원을 찾는 것이 더 가독성이 좋다고 판단했고, 그래서 이진탐색을 이용해서 문제를 해결하였습니다!

👀문제 해결 로직

1. 주어진 바위들의 위치를 오름차순으로 정렬한 후에, 도착 지점인 distance를 리스트에 추가합니다.

rocks.sort()
rocks.append(distance)

이렇게 하면, 마지막 바위에서 도착점까지의 거리도 계산에 포함됩니다!

2. 이분 탐색을 위한 left와 right를 설정하고, 가능한 최소 거리 mid를 설정하여 탐색을 진행합니다.

while left <= right:
    mid = (left+right) // 2

여기서 mid는 현재 검사하고자 하는 징검다리 사이의 최소 거리를 의미합니다.

3. 바위들을 순차적으로 확인하면서 현재 위차와 바위 위치 사이의 거리가 mid보다 작으면 바위를 제거하고, 그렇지 않으면 current를 해당 바위로 갱신합니다.

current = 0
removed_rocks = 0
for rock in rocks:
    if rock-current < mid:
        removed_rocks += 1
    else:
        current = rock

mid가 기준이 되는 이유는 문제에서 각 지점 사이의 거리의 최솟값 중에서 가장 큰 값을 구하라고 하였으므로, mid보다 작은 거리가 생기면 바위를 제거한 것으로 처리하고 그렇지 않으면 해당 바위를 갱신하는 것입니다. 즉 , 바위를 n개보다 많이 제거한 상황은 현재 설정한 mid가 너무 큰 값이라, 거리를 줄여서 바위를 덜 제거할 수 있도록 해야 한다는 뜻입니다.

4. 만약에 제거한 바위의 수가 제거할 바위의 수인 n보다 크다면, 너무 많은 바위를 제거한 것이므로 최소 거리를 줄이고 너무 많은 바위를 제거한 것이 아니라면 최소 거리를 늘립니다.

else:
    answer = mid
    left = mid+1

이러한 로직을 바탕으로 문제를 해결하였습니다!

👩‍💻최종 코드

def solution(distance, rocks, n):
    rocks.sort()
    rocks.append(distance)
    left, right = 1, distance
    
    while left <= right:
        mid = (left+right) // 2
        current = 0
        removed_rocks = 0
        
        for rock in rocks:
            if rock-current < mid:
                removed_rocks += 1
            else:
                current = rock
                
        if removed_rocks > n:
            right = mid-1
        else:
            answer = mid
            left = mid+1
        
    return answer

📚 새롭게 알게된 내용

[jung0115]

각 지점 사이의 거리의 최솟값 중에 가장 큰 값이라는 말을 바로 이해하기가 어려워서 살짝 혼란스러웠는데 혜원 님의 풀이를 보고 확실히 이해할 수 있었습니다 🥲

이번 차시도 수고하셨습니다!

class Solution {
    fun solution(distance: Int, rocks: IntArray, n: Int): Int {
        var answer = 0
        var rockClone = rocks.sortedArray() + distance
        
        var left = 1
        var right = distance
        
        while(left <= right) {
            val mid = (left + right) / 2
            
            var deleteRockCount = 0 // 제거한 바위의 수
            var position = 0
            
            rockClone.forEach { rock -> 
                if(rock - position < mid)
                    deleteRockCount++
                else
                    position = rock
            }
            
            if(deleteRockCount > n)
                right = mid - 1
            else {
                answer = mid
                left = mid + 1
            }
        }
        
        return answer
    }
}