洛谷P5650 字符串处理

题目描述
给定长度非0的非空01串SS。

找出SS的非空连续子串TT满足串中’0’的个数-'1’的个数最大。

你只需要输出最大值即可。

输入格式
一行一个01串表示SS。

输出格式
一行一个数表示答案。

输入输出样例
输入 #1 复制
0111100101
输出 #1 复制
2
说明/提示
设|S|=n

刚开始学算法的时候，我的做法是这样的 ：取1~len长度的字符串分别求和，比较大小，显然这时候的算法复杂度为o（n^2），无法得到满分。

暴力60分代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int main()
{
	int i=0,j=0;
	char a[100010];
	int b[100010];
	scanf("%s",a);
	int ans=0,bit=0;
	int len=strlen(a);
	for(i=0;i<len;i++)
	{
		if(a[i]=='0') bit=1;
		if(a[i]=='0') b[i]=1;
		if(a[i]=='1') b[i]=-1;
	}
	int len0=len;
	int m=0,n=0,cnt=0,t=0,cos=0;
	for(int i=1;i<=len0;i++)
	{
			int k=i;
			m=0,n=0;
			while(k!=0)
			{
				ans+=b[m];
				m++;
				k--;
			}
			
			while(m<len)
			{
				ans=ans-b[n]+b[m];
				n++,m++;
				cnt=max(cnt,ans);
			}
			int yy=i,plus=0;
			while(yy>0)
			{
				plus+=b[t];
				t++;
				yy--;
			}
			t=0;
			 cnt=max(cnt,plus);
			cos=max(cos,cnt);
			ans=0,cnt=0;
	}
	if(bit==0)
	{
		puts("-1");
		return 0;
	}
	printf("%d",cos);
	return 0;
	
}

现在用算法复杂度为o（n）的方法来解决它

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int main(){
	char s[100010];//取数组 
	cin>>s;
	int len=strlen(s);
	int ans=0,cnt=0;
	for(int i=0;i<len;++i){
		if(s[i]=='0')
			ans=max(++cnt,ans);//ans为历史最大值，cnt为当前值（即从上一个cnt被重置为0时算起） 
		else if(cnt<=0)	cnt=0;//当cnt小于0时对cnt进行重置 
		else cnt--;//当cnt大于0时减一 
	}
	if(ans==0)
		printf("-1");
	else
		printf("%d",ans);
	return 0;
}

ac代码如上。

这是最关键的两步

	ans=max(++cnt,ans);//ans为历史最大值，cnt为当前值（即从上一个cnt被重置为0时算起） 
		else if(cnt<=0)	cnt=0;//当cnt小于0时对cnt进行重置 

重置这一步代替了对不同长度字符串的处理，使字符串的长度变为动态的，从而使算法的复杂度得以降低