풍선 터트리기 - 월간 코드 챌린지 (python)

tjdud0123·2020년 9월 25일
1

프로그래머스 문제 풍선 터트리기 - LV3

https://programmers.co.kr/learn/courses/30/lessons/68646
시간복잡도를 줄이기 힘든 문제, 힙과 튜플사용

문제설명

일렬로 나열된 n개의 풍선이 있습니다. 모든 풍선에는 서로 다른 숫자가 써져 있습니다. 당신은 다음 과정을 반복하면서 풍선들을 단 1개만 남을 때까지 계속 터트리려고 합니다.

임의의 인접한 두 풍선을 고른 뒤, 두 풍선 중 하나를 터트립니다.
터진 풍선으로 인해 풍선들 사이에 빈 공간이 생겼다면, 빈 공간이 없도록 풍선들을 중앙으로 밀착시킵니다.
여기서 조건이 있습니다. 인접한 두 풍선 중에서 번호가 더 작은 풍선을 터트리는 행위는 최대 1번만 할 수 있습니다. 즉, 어떤 시점에서 인접한 두 풍선 중 번호가 더 작은 풍선을 터트렸다면, 그 이후에는 인접한 두 풍선을 고른 뒤 번호가 더 큰 풍선만을 터트릴 수 있습니다.

당신은 어떤 풍선이 최후까지 남을 수 있는지 알아보고 싶습니다. 위에 서술된 조건대로 풍선을 터트리다 보면, 어떤 풍선은 최후까지 남을 수도 있지만, 어떤 풍선은 무슨 수를 쓰더라도 마지막까지 남기는 것이 불가능할 수도 있습니다.

일렬로 나열된 풍선들의 번호가 담긴 배열 a가 주어집니다. 위에 서술된 규칙대로 풍선들을 1개만 남을 때까지 터트렸을 때 최후까지 남기는 것이 가능한 풍선들의 개수를 return 하도록 solution 함수를 완성해주세요.

입출력 예

a					result
[9,-1,-5]				3
[-16,27,65,-2,58,-92,-71,-68,-61,-33]	6

_솔루션 _

풍선을 남기는 것이 불가능한 경우는 왼쪽에 있는 풍선들의 최소값이 현재 풍선값 보다 작고,
오른쪽에 있는 풍선들의 최소값도 현재 풍선값 보다 작을 때이다.
맨 왼쪽의 풍선과 맨 오른쪽의 풍선은 항상 남는 것이 가능하므로 두번째 풍선부터 마지막에서 두번째 풍선까지 탐색을 한다.
시간 복잡도 문제때문에 heap을 사용하는데, 이 때 오른쪽에 있는 풍선들의 처리를 위해 idx를 함께 저장해준다.

# left[0][NUM] ,ballon, right[0][NUM]
-16 		27 	-92
-16 		65 	-92
-16 		-2 	-92
-16 		58 	-92
-16 		-92 	-71
-92 		-71 	-68
-92 		-68 	-61
-92 		-61 	-33

코드

# 파이썬
import heapq

def solution(a):
    balloons = [(b, i) for i, b in enumerate(a)]
    result = len(a)
    left, right = balloons[:1], balloons[1:]
    heapq.heapify(left)
    heapq.heapify(right)

    NUM, IDX = 0, 1
    for i, ballon in enumerate(a[1:-1], 1):
        if ballon == right[0][NUM]:
            while right[0][IDX] <= i:
                heapq.heappop(right)
        if ballon > left[0][NUM] and ballon > right[0][NUM]:
            result -= 1
        heapq.heappush(left, (ballon, i))
    return result

효율성 실패 코드

# 파이썬
import heapq

def solution(a):
    answer = len(a)
    for i, num in enumerate(a[1:-1], 1):
        left = min(a[:i])
        right = min(a[i+1:])
        if left < num and num > right:
            answer -= 1

    return answer

최대, 최소값을 구할 때 힙을 쓰면 시간복잡도를 줄일 수 있다.
이 문제에서는 왼쪽은 heappush를 이용하여 처리하기가 쉽지만, right(오른쪽 부분)을 처리하는게 조금 복잡하디.
최소값을 뽑을 때 현재 풍선보다 왼쪽에 있지만 아직 right저장되어 있는 풍선들이 문제가 되는데, 이 부분을 idx와 같이 저장하는 것으로 해결할 수 있다.

profile
Junior Web FE Developer

0개의 댓글