백준 알고리즘 2506번 점수계산(python)

단순 사칙연산 문제이다. 아래 코드 둘 다 정답이다.

N = int(input())
li = input().split('0')
res = 0
for s in li:
    cnt = s.count('1')
    if cnt > 0:
        res += cnt*(cnt+1) // 2
print(res)

N = int(input())
li = list(map(int, input().split()))
res = cnt = 0
for n in li:
    if n == 1:
        cnt += 1
    else:
        res += cnt*(cnt+1) // 2
        cnt = 0
res += cnt*(cnt+1) // 2
print(res)