18北邮计算机院-D.最长平衡子串 //哈希+贪心

最新推荐文章于 2024-08-21 10:13:22 发布

原创最新推荐文章于 2024-08-21 10:13:22 发布 · 716 阅读

4 ·

CC 4.0 BY-SA版权

北邮机试/OJ 同时被 2 个专栏收录

15 篇文章

订阅专栏

二分、贪心

10 篇文章

订阅专栏

本文详细介绍了一种高效算法，用于解决给定只含0和1的字符串中寻找最长平衡子串的问题。通过将0视为-1并计算前缀和，利用哈希表记录和值首次出现的位置，实现了快速查找最长平衡子串的长度。适用于字符串长度最大为10^6的情况。

题目描述
给定只含01的字符串，找出最长平衡子串的长度（平衡串：包含0和1的个数相同），串长最大10^6

输入

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

输出

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

示例输入

示例输出

2
0
4

思路：将0看做-1，求前i项和，当前i项和为0时，以此元素为结尾的最长长度为当前下标值+1；若该和值是第一次出现用hash数组记录该和值第一次出现时的下标i，否则当前下标值与hash数组中记录的当前和值第一次的下标值相减就是以此元素为结尾的最长长度。其中的最长的长度便是所要求的解。

实现代码：

#include<bits/stdc++.h>
using namespace std;
char s[1000003];
int h[2000003];
int main()
{
	int t,i,j;
	int m=1000001;//作为和值偏移值 防止出现负数无法用hash数组保存
	scanf("%d",&t);
	while(t--)
	{
	
		memset(h,-1,sizeof(h));
		scanf("%s",&s);
		int n=0,ma=0;
		for(i=0;i<strlen(s);i++)
		{
			if(s[i]=='0')n--;
			else n++;
			
			if(n==0)ma=max(ma,i+1);
			else if(h[n+m]==-1)h[n+m]=i;
			else {
				ma=max(ma,i-h[n+m]);	
			}
		}
		printf("%d\n",ma);
	}
}