백준 BAEKJOON 32642번 당구 좀 치자 제발 [PYTHON/파이썬]

 

 

 

 

 

 

백준 BAEKJOON 32642번 당구 좀 치자 제발 [PYTHON/파이썬]

---

<문제 출처> (BRONZE Ⅳ)

https://www.acmicpc.net/problem/32642

 

 

 

 

 

 

 

<풀이>

동우의 현재 분노를 0으로 지정해 준 뒤

날씨의 정보를 확인하여 순회하여 분노 값을 리스트에 담았다.

그리고 sum을 사용하여 분노의 합을 출력하였다.

 

 

 

 

 

 

 

 

<코드>

N = int(input())
weather = list(map(int, input().split()))
anger = [0] * N  # 각 날 마다의 분노 리스트
status = 0  # 동우의 현재 분노 0

for i in range(N):
    if weather[i] == 1:
        status += 1
    elif weather[i] == 0:
        status -= 1

    anger[i] = status

print(sum(anger))