牛客小d和孤独的区间

牛客小d和孤独的区间

这道题我们如果等到数据都输入完后再计算会浪费很多时间

所以还是每次输入都计算一遍

我们知道要找到一个区间里只有一个1,其余都是0

所以这时候我们可以将输入分为两种情况

1,输入为1
2,输入为0

这时我们去想,如果是1的话,可以和这个1前面的0组合,有多少个0就有多少个0加1本身个区间

如果是0,则包括与前一个1在内所有的0,与前一个1前的0进行组合,

那么记录下来当前有多少个0和前一个1前有多少个0就可以了

#include <bits/stdc++.h>
using namespace std;
int main(){
    int n;
    cin >> n;
    long long temp,qq,cc,ans;
    ans=0;
//     qq是最近的一前面有多少个0
    qq=0;
//     cc前面有多少个0
    cc=0;
    for(int i = 0;i < n;i++){
        cin >> temp;
        if(temp == 0)
        {
            
            ans+=qq;
            cc++;
        }
        else if(temp == 1)
        {
            ans+=cc+1;
            qq=cc+1;
            cc=0;
        }
    }
    cout << ans;
}