結果

問題 No.1765 While Shining
コンテスト
ユーザー roaris
提出日時 2021-12-09 16:45:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 97 ms / 2,000 ms
コード長 472 bytes
コンパイル時間 169 ms
コンパイル使用メモリ 82,388 KB
実行使用メモリ 103,628 KB
最終ジャッジ日時 2024-07-17 09:30:39
合計ジャッジ時間 3,424 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N = int(input())
A = list(map(int, input().split()))
i = 0
ans = 0

while i<N:
    if A[i]==0:
        i += 1
    else:
        j = i
        
        while j<N:
            if ((j-i)%2==0 and A[j]==0) or ((j-i)%2==1 and A[j]==1):
                break
            j += 1
        
        l = j-i
        
        if j==N:
            l -= 1
        
        for k in range(l, -1, -2):
            ans += k

        i = j

print(ans)
0