[python] 코딩테스트 대비 - 점수계산

문제)

 

 

 

답안 코드)

import sys
#sys.stdin=open("input.txt","rt")
N=int(input())
box=list(map(int,input().split()))
weight=1
score=0
for i in box:
    if i==1:#맞췄을때
        score+=weight*i
        weight+=1
    else:#맞추지 못했을때
        weight=1
print(score)