結果

問題 No.1765 While Shining
ユーザー atjh16
提出日時 2021-12-27 11:41:33
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 55 ms / 2,000 ms
コード長 682 bytes
コンパイル時間 1,661 ms
コンパイル使用メモリ 192,516 KB
最終ジャッジ日時 2025-01-27 07:21:07
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

int n;
bool a[200000];
long long m = 0;
long long b[200000];

int main(){
    cin >> n;
    
    for(int i = 0; i < n; i++)
        cin >> a[i];
    
    for(int i = n - 2; i >= 0; i--){
        if(a[i]){
            bool f = 0;
            int j = i + 1;
            b[i]++;
            
            while(j < n - 1 && a[j] == f){
                if(b[j] > 0){
                    b[i] += b[j];
                    break;
                }
                
                b[i]++;
                f = !f;
                j++;
            }
                        
            m += b[i];
        }
    }
    
    cout << m << endl;
}
0