백준(Python) 2506번 점수계산 풀이

Python으로 구현한 2506번 점수계산 문제 풀이입니다.

 

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

2506번: 점수계산

---

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

count = 0
score = 0
for i in range(n) :
  if data[i] == 1 :
    count += 1
    score += count
  else :
    count = 0

print(score)

 

반복문을 통해 data 리스트의 값들을 하나씩 확인하여 그 값이 1이라면 count를 1증가시킨 뒤 점수에 해당되는

 

score에 count 값을 누적한다. 만약 data[i]의 값이 0이라면 count 값을 다시 0으로 초기화해준다.

 

반복문이 종료되면 최종적으로 score 값을 출력한다.