题目:近来,蒜头君致力于研究班上同学的配对问题(别想太多,仅是舞伴),通过各种推理和实验,他掌握了大量的实战经验。例如,据他观察,身高相近的人似乎比较合得来。
万圣节来临之际,蒜头君准备在学校策划一次大型的“非常男女”配对活动。对于这次活动的参与者,蒜头君有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单。他让学校的所有人按照身高排成一排,然后从中选出连续的若干个人,使得这些人中男女人数相等。为了使活动更热闹,蒜头君当然希望他能选出的人越多越好。请编写程序告诉他,他最多可以选出多少人来
一个小技巧:对于只有两种数的情况,我们令其分别为-1和+1 ,前缀和sum[i]=x,这样的sum[i]除了单纯的前缀和之外还有个很重要的性质:
1. 若sum[i]=0则代表-1和+1相同
2.若sum[i]>0则代表-1比+1少了sum[i]个
3.若sum[i]<0则代表-1比+1多了sum[i]个
所以只要两个地方的前缀和是相同的,那么这两个地方之间的区域一定满足-1和+1个数相同,具体见AC代码:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <map>
#include <algorithm>
#define int long long
using namespace std;
const int inf=0x3f3f3f3f;
const int N=1e6+5;
int s[N];
main(){
int n;
cin>>n;
int ans=0;
for(int i=1;i<=n;i++){
int x;
cin>>x;
if(x)s[i]=s[i-1]+1;
else s[i]=s[i-1]-1;
if(!s[i])ans=max(ans,i);
}
map<int,int>mp;
for(int i=1;i<=n;i++){
if(!mp.count(s[i]))mp[s[i]]=i;
else ans=max(ans,i-mp[s[i]]);
//cout<<i<<" "<<s[i]<<" "<<mp[s[i]]<<endl;
}
cout<<ans<<endl;
}