近来,蒜头君致力于研究班上同学的配对问题(别想太多,仅是舞伴),通过各种推理和实验,他掌握了大量的实战经验。例如,据他观察,身高相近的人似乎比较合得来。
万圣节来临之际,蒜头君准备在学校策划一次大型的“非常男女”配对活动。对于这次活动的参与者,蒜头君有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单。他让学校的所有人按照身高排成一排,然后从中选出连续的若干个人,使得这些人中男女人数相等。为了使活动更热闹,蒜头君当然希望他能选出的人越多越好。请编写程序告诉他,他最多可以选出多少人来。
输入格式
第一行有一个正整数 n (1≤n≤105),代表学校的人数。
第二行有 n 个用空格隔开的数,这些数只能是 0 或 1,其中,0 代表一个女生,1代表一个男生。
输出格式
输出一个非负整数,这个数表示在输入数据中最长的一段男女人数相等的连续子序列长度。
输出时每行末尾的多余空格,不影响答案正确性
样例输入
9
0 1 0 0 0 1 1 0 0
样例输出
6
这道题属于前缀和(前缀和是一个数组的某项下标之前(包括此项元素)的所有数组元素的和。)算法的应用,但是我一开始没想那么多,以为就直接对输入的男女0和1分别计数,然后输出少的那一个的2倍就行了,后来测试数据过了3组。。。发现没那么简单,要求输出最长的一段男女人数相等的连续子序列长度,重点要连续。
在原数组中,我们可以用-1代表女生,1代表男生,然后问题就转化为求一个最长子序列和为0的长度了。因为一个男生加一个女生正好为0,只要找到两个位置的前缀和相等,他们之间就是男女相等的连续子序列。我们记录每个前缀和第一次出现的位置,下次出现的时候,减去第一次的位置就是长度。要注意:前缀和都是从数组的第二个位置开始存储,也就是下标为 1 的位置开始存储,为了方便计算下标为 0 的位置都是 0 。第一重循环规定所求的区间长度,第二重循环来求该区间最长的子序列。只要找到相等就 break 直接求下一个区间长度的最大子序列,因为我们是从小到大开始遍历的,所以在所求区间中第一次出现相等就代表最长的子序列,后面的长度就没有必要进行求