题目大意是给你一串序列,序列只有01两种数,选取一个区间,这个区间的值为0的个数减去1的个数,问这个序列能组成的不同值的区间有多少种?序列长度1e5,这道题说实话想了很长时间,不知道怎么降复杂度,后来看题解才明白,最重要的是只有01两种数,所以我们计算前缀和,算出的肯定是连续的区间,比如说{1,1,1,1,0,0,1}算出的前缀和是{-1,-2,-3,-4,-3,-2,-3},中间的每个数肯定都是连续的,所以我们可以找到这个区间能组成的最大的值和最小的值,然后相减就行,因为能组成的最大值的区间,肯定往左或往右扩展一个,肯定就会变小。这里注意区间可以是空集,所以肯定会有0,最后相减要+1
#include<bits/stdc++.h>
using namespace std;
using LL = int64_t;
const int maxn=2e5+5;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const LL LLINF=1e14;
const double pi=acos(-1.0);
int node[maxn],n;
int find_max() {
int maxs=0,sum=0;
for(int i=1;i<=n;i++) {
if(node[i]==0) sum++;
else sum--;
if(sum>maxs) maxs=sum;
if(sum<0) sum=0;
}
return maxs;
}
int find_min() {
int mins=0,sum=0;
for(int i=1;i<=n;i++) {
if(node[i]==0) sum++;
else sum--;
if(sum<mins) mins=sum;
if(sum>0) sum=0;
}
return mins;
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cin>>n;
for(int i=1;i<=n;i++) cin>>node[i];
cout<<find_max()-find_min()+1<<endl;
return 0;
}