51nod oj 1393 0和1相等串 【贪心】

题目链接：1393

1393 0和1相等串

基准时间限制：1 秒 空间限制：131072 KB 分值: 20  难度：3级算法题

 收藏

 关注

给定一个0-1串，请找到一个尽可能长的子串，其中包含的0与1的个数相等。

Input

一个字符串，只包含01，长度不超过1000000。

Output

一行一个整数，最长的0与1的个数相等的子串的长度。

Input示例

1011

Output示例

2

开始的时候一直想着记录从起点到每个位置0和1的个数---然后就不会写了--一直写不出来

今天突然想到的---我们只需要记录一下从起点开始到每个位置的0和1个数的差值---

一个相等子串的开头前面那个位置和结尾的那个位置的0和1的差值一定是一样的====

排序就解决问题了---

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct node{
	int cha;
	int hao;
}dian[1000100];
bool cmp(node xx,node yy)
{
	if (xx.cha!=yy.cha)
	return xx.cha<yy.cha;
	return xx.hao<yy.hao;
}
char ch[1000100];
int main()
{
	scanf("%s",ch);
	int ll=strlen(ch);
	dian[0].hao=0;
	dian[0].cha=0;
	for (int i=0;i<ll;i++)
	{
		dian[i+1]=dian[i];
		if (ch[i]=='1')
		dian[i+1].cha++;
		else
		dian[i+1].cha--;
		dian[i+1].hao=i+1;
	}
	sort(dian,dian+ll+1,cmp);
	int ans=0,qian=0;
	for (int i=1;i<=ll;i++)
	{
		if (dian[i].cha==dian[qian].cha)
		ans=max(ans,dian[i].hao-dian[qian].hao);
		else
		qian=i;
	}
	printf("%d\n",ans);
	return 0;
}