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;
}