[인프런 - 파이썬 알고리즘 문제풀이] 2.10 점수계산

인프런 > 파이썬 알고리즘 문제풀이

섹션 2.10 점수계산

👊 내 문제 풀이

앞서 풀었던 백준 8958번(OX퀴즈) 문제와 거의 동일한 문제다.

정답일 경우(1) 총 점수에 누적점수를 더한 한 후 누적점수에 1점을 추가하고, 오답일 경우(0) 누적점수를 1로 초기화해준다.

score=0		#총점수
plusscore=1	#누적점수
N=int(input())
arr=list(input())
for i in arr:
	if i == "1":
		score += plusscore
		plusscore += 1
	elif i == '0':
		plusscore = 1
print(score)

채점 결과

 

💡 강의 해답

n=int(input())
a=list(map(int, input().split()))
cnt=0
sum=0
for i in range(n):
    if a[i]==1:
        cnt=cnt+1
        sum=sum+cnt
    else:
        cnt=0

print(sum)