CF1352 E.Special Elements

CF1352E

Question

给定数组$a$满足$1\le a[i]\le n(1\le n\le 8000)$，若数组中任意连续且长度$\ge 2$的和等于$a[i]$则贡献值$+1$，求最后贡献值为多少？

Solution

数据范围比较小直接$O(N^2)$暴力即可。

写这篇题解是想告诉自己同样是判断状态的真假，不同的容器相差的时间很多。
比如这道题用set就会被卡，用unordered_set就280ms左右能过，用数组表示状态就46ms。
unordered_set猜测是哈希映射时间导致的比较慢，set基于红黑树有一个$O(logN)$就再慢一点。
根据情况选择合适的数据结构也是非常重要的。

Code

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int t,n,a[8005];
bool flag[8005];
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)scanf("%d",&a[i]),flag[i]=0;
		for(int i=1;i<=n;i++)
		{
			int s=a[i];
			for(int j=i+1;j<=n;j++)
			{
				s+=a[j];
				if(s<=n)flag[s]=1;
			}
		}
		int ans=0;
		for(int i=1;i<=n;i++)
		  if(flag[a[i]])ans++;
		printf("%d\n",ans);
	}
	return 0;
}