점수계산

it 취업을 위한 알고리즘 문제 풀이

문제

코드

import sys
sys.stdin = open("input.txt","rt")

n = int(input())
OX = list(map(int,input().split()))
cnt = 0
score = 0

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

print(score)

문제해결방법

• OX 리스트에 1인지 아닌지 확인하면서 연속된 1을 수를 cnt에 누적한다.
• 누적된 수를 점수에 더하고 값을 누적한다.
• 리스트에 0인 값을 만나면 다시 누적된 수 cnt를 0으로 초기화 하고 위 과정을 다시 진행한다.