题目
小b有一个01序列,她想找到一个最长的区间使得这个区间的01能两两配对,即0的个数和1的个数相等。求最长区间的长度。
输入
第一行一个正整数n,表示数组长度,其中0<n≤50000;
第二行n个0或1,以空格隔开。输出
输出一个数,表示最长区间的长度
输入样例
3
0 1 0输出样例
2
思路:
由于要求最长区间使得 0、1 能够两两配对,那么可以将 0 转成 -1,这样 -1、1 相加进行枚举,题目就转化成求最长连续的 0 个个数
故先进行预处理求前缀和,然后枚举前缀和看是否为 0 即可
源程序
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define EPS 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
const int MOD = 1E9+7;
const int N = 100000+5;
const int dx[] = {0,0,-1,1,-1,-1,1,1};
const int dy[] = {-1,1,0,0,-1,1,-1,1};
using namespace std;
int sum[N];
int main() {
int n;
scanf("%d",&n);
for(int i=1; i<=n; i++) {
int x;
scanf("%d",&x);
if(x==0)//0转为-1
sum[i]=sum[i-1]-1;
else
sum[i]=sum[i-1]+1;
}
int res=-1;
for(int i=1; i<n; i++) {
for(int j=i+1; j<=n; j++) {
if(sum[j]-sum[i-1]==0)
res=max(res,j-i+1);
}
}
printf("%d\n",res);
return 0;
}