백준 11509 풍선 맞추기

24.56m

나의 답

import sys
input = sys.stdin.readline

n = int(input().rstrip())
h = list(map(int, input().rstrip().split()))
stack = []

for i in range(n):
    is_push = False
    for s in stack:
        if s[-1] == h[i] + 1:
            s.append(h[i])
            is_push = True
            break
    if not is_push:
        stack.append([h[i]])

print(len(stack))

아이디어만 찾으면 생각보다 쉽게 풀리는 문제였다!
처음에 스택을 탐색할 때 len(stack) 돌렸더니 시간초과 났는데
스택 자체를 탐색하니까 시간초과도 안나고 잘 됐다.