[BOJ/백준] 32642 - 당구 좀 치자 제발 (Python)

문제

문제링크

현재 동우의 분노는 0이고 비가 오면(1) 1씩 증가하고 오지 않으면(0) 1씩 감소한다.

각 날의 분노를 구해 총 합을 구하라.

 

풀이

anger을 0으로 설정하고 입력받은 값이 1이면 1을 더하고 아니라면 1을 뺀다.

총합(gauge)에 각 날의 anger을 더하고 for문이 끝날 때까지 이 과정을 반복한다.

 

코드

n=int(input())
rain=list(map(int, input().split()))

anger,gauge=0,0
for r in rain:
    if r==1:
        anger+=1
    else:
        anger-=1
    gauge+=anger
print(gauge)