풍선 터뜨리기

문제 설명

일렬로 나열된 n개의 풍선이 있습니다. 모든 풍선에는 서로 다른 숫자가 써져 있습니다. 당신은 다음 과정을 반복하면서 풍선들을 단 1개만 남을 때까지 계속 터트리려고 합니다. 임의의 인접한 두 풍선을 고른 뒤, 두 풍선 중 하나를 터트립니다. 터진 풍선으로 인해 풍선들 사이에 빈 공간이 생겼다면, 빈 공간이 없도록 풍선들을 중앙으로 밀착시킵니다. 여기서 조건이 있습니다. 인접한 두 풍선 중에서 번호가 더 작은 풍선을 터트리는 행위는 최대 1번만 할 수 있습니다. 즉, 어떤 시점에서 인접한 두 풍선 중 번호가 더 작은 풍선을 터트렸다면, 그 이후에는 인접한 두 풍선을 고른 뒤 번호가 더 큰 풍선만을 터트릴 수 있습니다. 당신은 어떤 풍선이 최후까지 남을 수 있는지 알아보고 싶습니다. 위에 서술된 조건대로 풍선을 터트리다 보면, 어떤 풍선은 최후까지 남을 수도 있지만, 어떤 풍선은 무슨 수를 쓰더라도 마지막까지 남기는 것이 불가능할 수도 있습니다. 일렬로 나열된 풍선들의 번호가 담긴 배열 a가 주어집니다. 위에 서술된 규칙대로 풍선들을 1개만 남을 때까지 터트렸을 때 최후까지 남기는 것이 가능한 풍선들의 개수를 return 하도록 solution 함수를 완성해주세요.

 

풀이

def solution(a):   
    
    cnt = 0
    for i in range(len(a)):
        if min(a[:i+1]) >= a[i] or min(a[i:]) >= a[i]:
            cnt += 1

    return cnt
    
    # 좌우에 자기보다 작은 게 존재하면 살아남지 못한다. 
    # not ( min(a[:i+1]) < a[i] and min(a[i:]) < a[i] )
    # 좌우에 자기보다 큰 게 하나라도 존재하면 된다. / 좌우 한 쪽이라도 자기가 최소면 된다
    #      min(a[:i+1]) >= a[i] or min(a[i:]) >= a[i]

시간 초과로 실패했다. 논리로 볼 때 큰 이상은 없는 것 같아서, 시간을 줄일 수 있는 방법을 찾아야할 것 같다.

min() 함수를 쓰면 시간이 오래 걸리므로 쓰지 않는 방법을 찾아야했다.

 

한 블로그를 보니 재미있는 코드를 짰다.

def solution(a):
    answer = [False for i in range(len(a))]
    INF = 1000000001 # a의 모든 수는 -1,000,000,000 이상 1,000,000,000 이하인 정수 조건을 교묘히 피한다
    left_min, right_min = INF, INF #끝 값을 무조건 True로 만들어주기 위해
    
    for i in range(len(a)):  
        if a[i] < left_min:  # 왼쪽에 더 작은 값이 있는지 확인
            left_min = a[i]  # 왼쪽 최솟값 갱신
            answer[i] = True
        if a[len(a)-i-1] < right_min:  # 오른쪽부터 최솟값 갱신을 위해 ! 
            right_min = a[len(a)-i-1]  # 마치 테넷을 본 기분이다
            answer[len(a)-i-1] = True
    return sum(answer)

아래의 if 코드가 너무 대단하다. 어떻게 a[ len(a) - i - 1 ] 를 써서 answer[ i ] 를 좌우 한번에 최솟값 갱신을 보일 생각을 했을까. 배워갈 점이다.

 

코드를 짜면 어떻게든 실행은 되겠지만 제한 시간 안에 풀기 위해 여러 방법을 모색하는 게 프로그래밍의 재미인 것 같다. 

 

 

출처

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

프로그래머스