CF1352E
Question
给定数组 a a a满足 1 ≤ a [ i ] ≤ n ( 1 ≤ n ≤ 8000 ) 1\le a[i] \le n(1\le n\le 8000) 1≤a[i]≤n(1≤n≤8000),若数组中任意连续且长度 ≥ 2 \ge 2 ≥2的和等于 a [ i ] a[i] a[i]则贡献值 + 1 +1 +1,求最后贡献值为多少?
Solution
数据范围比较小直接 O ( N 2 ) O(N^2) O(N2)暴力即可。
写这篇题解是想告诉自己同样是判断状态的真假,不同的容器相差的时间很多。
比如这道题用set就会被卡,用unordered_set就280ms左右能过,用数组表示状态就46ms。
unordered_set猜测是哈希映射时间导致的比较慢,set基于红黑树有一个
O
(
l
o
g
N
)
O(logN)
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;
}