結果
| 問題 | 
                            No.1505 Zero-Product Ranges
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2022-01-02 18:12:23 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                TLE
                                 
                             
                            
                            (最新)
                                AC
                                 
                             
                            (最初)
                            
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 607 bytes | 
| コンパイル時間 | 351 ms | 
| コンパイル使用メモリ | 82,344 KB | 
| 実行使用メモリ | 75,268 KB | 
| 最終ジャッジ日時 | 2024-10-11 22:03:50 | 
| 合計ジャッジ時間 | 13,509 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge1 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 48 TLE * 1 | 
ソースコード
N = int(input())
A = input().replace(' ', '')
all = N * (N + 1) // 2
def search_one():
    a = 0
    b = len(A)
    while True:
        j = a + (b - a) // 2
        if '1' * j in A:
            a = j
        else:
            b = j
        if b - a < 3:
            break
    return b
i = N
onecnt = 0
while i > 0:
    i = search_one()
    found = False
    while found == False and i > 0:
        o = A.count('1' * i)
        if o > 0:
            onecnt += (i * (i + 1) // 2) * o
            A = A.replace('1' * i, '')
            found = True
        else:
            i -= 1
    
print(all - onecnt)