B. Balanced Substring

传送门
题意:

kele妹妹喜欢黄色, 但是若愚姐姐喜欢白色, 她俩谁也说服不了对方, 于是她俩决定在颜色上面要做到均衡, 即两种颜色一定范围内出现的数量要相同才可以
现在用0代表kele妹妹喜欢的黄色,用1代表若愚姐姐喜欢的白色
给定一个由1和0组成的字符串, 希望找到一个最长的子串, 红色和白色的数量一致

思路:

因为是要找到一个最长的子串,并且其中的1和0的数量相同,那么可以把0当做-1处理,那么做前缀处理就是出现的前缀相同的时候两个范围内的区间是满足条件的,遍历一遍取max即可。

代码:


#include<cstdio>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <string>
#include <math.h>
#include<vector>
#include<queue>
#include<map>
#define sc_int(x) scanf("%d", &x)
#define sc_ll(x) scanf("%lld", &x)
#define pr_ll(x) printf("%lld", x)
#define pr_ll_n(x) printf("%lld\n", x)
#define pr_int_n(x) printf("%d\n", x)
#define ll long long 
using namespace std;

const int N=1000000+100;
int n ,m,h;
char s[N];
int dp[N];

void slove( )
{
	sc_int(n);
	cin>>s+1;
	for(int i =1;i<=n;i++)
	{
		if(s[i]=='1') dp[i]=dp[i-1]+1;
		else dp[i]=dp[i-1]-1;
	}
	map<int,int>q;
	int res=0;
	q[0]=0;
	for(int i =1;i<=n;i++)
	{
		if(!q[dp[i]])q[dp[i]]=i;//只要纪录第一次出现的下标即可
		else res=max(res,i-q[dp[i]]);

		if(dp[i]==0)res=max(res,i);//因为如果为0的话,那么整个前缀和的长度都可以满足条件
	}
	cout<<res;
	cout<<endl;
}

int main()
{
	int t;
	// sc_int(t);
	// while(t--)
	slove();
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值