【BHOJ 最长零子数组】前缀和 | 思维 | STL | E

【BHOJ 420】最长零子数组

时间限制: 1000 ms 内存限制: 131072 kb 总通过人数: （未加载）总提交人数: （未加载）

Tags：前缀和 STL

题目描述

给定一个仅含 +1、-1 的数组，问其最长和为零的子数组（连续）长度是多少？

输入

多组，以 $EOF$ 结束。

每组两行，第一行一个正整数 $n$ 为数组长度，第二行 $n$ 个数，只会为 $\pm 1$。

范围： $1\le n\le 5\times 10^6$。

输出

对每组数据输出一个数，表示其最长和为零的子数组（连续）的长度。

输入样例

10
1 -1 1 1 -1 -1 -1 1 1 1

输出样例

8

分析

（本题改编自猿题库2016校招笔试）

这道题和 【BCPC2015 偶回文串】有异曲同工之妙，算是它的简化版（因为只用求最长段长度而不用求一共有多少对）

也是巧妙运用 前缀和 的一道题：

（即利用前缀和表示状态，然后对于状态相同的每个点对，其中间区段就具有某种抵消的特性。然后对于每个点对都求出对应的解，维护全局最值即可得解）

• 在线处理。读第 $i$ 个数之后，算出当前前缀和 $S$
• 然后看有没有曾经遇到过这个前缀和：
  • 若没遇见过，则保存当前下标（此时的下标就是第一次出现前缀和 $S$ 的下标）（不妨用 $std::unordered\_map$）
  • 若遇见过，则拿当前的下标 $i$ 减去第一次出现这个前缀和的下标，然后拿这个 $\Delta i$ 更新最大长度。

这样，读完数据后答案就出来了。

（另外别忘了在读数之前往 $std::unordered\_map$ 里面插入 $(0,-1)$，即表示什么也不加的、前缀和为 $0$ 的情况）

时间复杂度：

• std::unordered_map，查询插入都近似 $O(1)$ 的，总操作就是 $O(N)$ 的。

AC代码

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unordered_map>

#define GC getchar()
#define PC putchar
#define _Q_(x) {register char _c=GC,_v=1;for(x=0;_c<48||_c>57;_c=GC)if(_c==45)_v=-1;
#define _H_(x) for(;_c>=48&&_c<=57;x=(x<<1)+(x<<3)+_c-48,_c=GC);if(_v==-1)x=-x;}
#define sc(x) _Q_(x)_H_(x)
#define se(x) _Q_(x)else if(_c==EOF)exit(0);_H_(x)

template<class T>
void UPRT(const T a){if(a>=10)UPRT(a/10);PC(a%10+48);}

std::unordered_map<int, int> mp;
int main()
{
	int n, tp;
	while (1)
	{
		se(n)
		mp.clear();
		if(n == 1)
		{
			sc(tp)
			PC('0'), PC(10);
			continue;
		}
		int sum = 0, ans = 0;
		mp.insert(std::make_pair(0, -1));
		for (int i=0; i<n; ++i)
		{
			sc(tp)
			if (tp+1) ++sum;
			else --sum;
			auto it = mp.find(sum), E = mp.end();
			if (it != E)
			{
				int d = i-it->second;
				if (d > ans)
					ans = d;
			}
			else
			{
				mp.insert(std::make_pair(sum, i));
			}
		}
		UPRT(ans), PC(10);
	}
}