题目描述

小Q面前的桌子上有$N$个硬币，$0$表示正面，$1$ 表示反面，现在他有一次机会可以选择一个$a_i$~$bi$的段，把这个段的硬币都翻转一面，他现在想知道$N$个硬币中最多可以有多少个硬币正面朝上。

输入

第一行一个整数$N$表示桌子上有$N$个硬币。 第二行为$N$个$0$和$1$，表示硬币$i$的状态。其中$0$表示正面， $1$表示反面。

输出

第一行有一个整数，表示翻转后最多有多少个硬币正面朝上。

输入样例1

4
1 0 1 1

输出样例1

3

输入样例2

7
0 1 1 0 1 1 0

输出样例2

6

提示

【数据范围】

$30$的数据 $1<=N<=100$

$60$的数据 $1<=N<=10^4$

$100$的数据$1<=N<=10^6$

我的代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
int n,a[1000003],b[1000003],l,r=1,sum,ans;
int main(){ 
    std::ios::sync_with_stdio(0);
    cin>>n;l=n;
    for(int i=1;i<=n;i++)
    {
    	cin>>a[i];
    	if(a[i]==0)a[i]=1,sum++,ans++;
    	else a[i]=0;
    	b[i]=b[i-1]+a[i];
	}
	for(int i=n;i>0;i--)
		if(a[i]==0){l=i;break;}
	for(int i=1;i<=n;i++)
	    if(a[i]==0){r=i;break;}
	for(int i=1;i<=l;i++)
	{
		int m=l-i+1,x=b[l]-b[i-1],y=m-x;//x:正面,y:反面 
		ans=max(ans,sum-x+y);
	}
	for(int i=n;i>=r;i--)
	{
		int m=i-r+1,x=b[i]-b[r-1],y=m-x;
		ans=max(ans,sum-x+y);
		//cout<<x<<" "<<y<<endl;
	}
	cout<<ans;
    return 0;
}

不知道Where错了？试了几个数据都没错。