1410: PIPI的开关（尺取法）

题目描述

PIPI有编号为1-n的开关，每个开关0表示关闭，1表示开启。
一开始，每个开关都有自己的初始状态。PIPI有一次机会，可以选择一个任意的区间[L,R]，将该区间内的所有开关都按一次(即0->1,1->0)。
请问开关开启的数量最多是多少？

输入

第一行输入一个正整数n，表示开关数量，n<=10^7。
接下来一行一个01串，表示1到n号开关的初始状态。

输出

输出开关开启的最多数量。

样例输入

4
1001

样例输出

4

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+5;
int sum[N];
char s[N];
int main(){
	int n;
	scanf("%d",&n);
	scanf("%s",s+1);
	for(int i=1;i<=n;i++){
		sum[i]=sum[i-1]+s[i]-'0'; ///求前缀和，代表前i个位置中1的个数	
	}
	int ans=sum[n],l,r;
	for(l=1,r=1;r<=n;r++){
		if(2*(sum[r]-sum[l-1])>=r-l+1) l=r+1; ///如果此区间中1的个数大于等于0的个数，左指针等于r+1 
		else ans=max(ans,sum[n]+r-l+1-2*sum[r]+2*sum[l-1]); ///更新ans的值 
	}
	printf("%d\n",ans);
}