[Python] BOJ 11509번. 풍선 맞추기

11509번. 풍선 맞추기

문제 링크

풀이 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# 11509번. 풍선 맞추기


import sys
input = sys.stdin.readline

N = int(input())
nList = list(map(int, input().split()))
# 이 높이에서 날아오는 화살이 있나 check
arrow = [0 for height in range(1000001)]
ans = 0

for n in nList:
    # n+1 높이에서 날아오는 화살이 없는 경우
    if arrow[n+1] == 0:
        # n 높이에서 날아오는 화살 추가
        arrow[n] += 1
        ans += 1
    # n+1 높이에서 날아오는 화살이 있는 경우
    else:
        # n 높이로 교체
        arrow[n+1] -= 1
        arrow[n] += 1
print(ans)

비고