프로그래머스 풍선 터트리기 파이썬(python)

문제

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

1. 임의의 인접한 두 풍선을 고른 뒤, 두 풍선 중 하나를 터트립니다.
2. 터진 풍선으로 인해 풍선들 사이에 빈 공간이 생겼다면, 빈 공간이 없도록 풍선들을 중앙으로 밀착시킵니다.

여기서 조건이 있습니다. 인접한 두 풍선 중에서 번호가 더 작은 풍선을 터트리는 행위는 최대 1번만 할 수 있습니다. 즉, 어떤 시점에서 인접한 두 풍선 중 번호가 더 작은 풍선을 터트렸다면, 그 이후에는 인접한 두 풍선을 고른 뒤 번호가 더 큰 풍선만을 터트릴 수 있습니다.

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

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

제한 사항

• a의 길이는 1 이상 1,000,000 이하입니다.
• a [i]는 i+1 번째 풍선에 써진 숫자를 의미합니다.
• a의 모든 수는 -1,000,000,000 이상 1,000,000,000 이하인 정수입니다.
• a의 모든 수는 서로 다릅니다.

---

풀이

[처음 접근 한 풀이 방법]

1. 번호가 더 작은 것을 고르는 기회는 한번뿐이다.
2. 이 한 번을 제외하곤 무조건 큰 것을 골라야 한다.
3. 왼쪽, 오른쪽 각각 최소 값을 경신해서 저장해 두면 풀리지 않을까?

def solution(a):
    answer = 0
    min_dp = [float('inf')] * len(a)
    min_dp[0] = a[0]
    # 왼쪽에서 최솟값 갱신하면서 저장해둔다.
    for i in range(1, len(a)):
        min_dp[i] = min(min_dp[i-1], a[i])

	# 오른쪽부터 탐색을 시작하고 오른쪽 기준해서 최소 값을 갱신해둔다.
    # 현재 풍선 기준 양 쪽이랑 값 비교했을 때 한번은 커도 된다. 두번은 x
    # 풍선 list 양 끝은 무조건 살아남음 -> 마지막에 + 2 해주는 이유
    min_dp[-1] = a[-1]
    for i in range(len(a)-2,0, -1):
        # a[i]값이 왼, 오 둘 다보다 작다면 통과
        if a[i] < min_dp[i-1] and a[i] < min_dp[i+1]:
            answer += 1
        # a[i]값이 왼, 오 하나보단 큰 경우도 통과
        elif (min_dp[i - 1] < a[i] < min_dp[i + 1]) or (min_dp[i - 1] > a[i] > min_dp[i + 1]):
            answer += 1
        # a[i]값이 왼, 오 둘 다 보다 큰 경우는 패스
        else:
            pass
        # 오른쪽 기준 최소 값은 계속 갱신해줘야된다.
        min_dp[i] = min(min_dp[i+1], a[i])
    return answer + 2

[새로 접근한 풀이]

코드는 통과되지만 걸리는 시간은 처참하다. 아래는 다른 블로그를 참고하여 two point 이용하여 푼 풀이방법이다.

1. 앞, 뒤 동시에 최소 값이 경신되는 지점을 저장한다.
2. 탐색을 마친 뒤 갱신 지점들의 개수가 answer가 된다.

def solution(a):
    result = [False] * len(a)
    min_front = float('inf')
    min_back = float('inf')
	# 앞, 뒤 동시에 최소 값이 갱신되는 지점을 저장해둔다.
    # 갱신 지점들의 개수가 answer가 된다.
    for i in range(len(a)):
        if a[i] < min_front:
            min_front = a[i]
            result[i] = True
        if a[-1-i] < min_back:
            min_back = a[-1-i]
            result[-1-i] = True
    return sum(result)

 

[참고한 블로그]

https://velog.io/@eehwan/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%ED%92%8D%EC%84%A0-%ED%84%B0%ED%8A%B8%EB%A6%AC%EA%B8%B0-%ED%8C%8C%EC%9D%B4%EC%8D%AC

---

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/68646

 

프로그래머스