[codeforces 1352E] Special Elements 桶排序+前缀和

Codeforces Round #640 (Div. 4)   参与排名人数9749,终于弄明白账号前*的意义,*out of competition,也即虽然该用户参加本场比赛,但不参与排名。

[codeforces 1352E]   Special Elements   桶排序+前缀和

总目录详见https://blog.csdn.net/mrcrack/article/details/103564004

在线测评地址http://codeforces.com/contest/1352/problem/E

Problem	Lang	Verdict	Time	Memory
E - Special Elements	GNU C++17	Accepted	46 ms	3700 KB

可用前缀和，计算出任意一段连续数组元素的和，看看是否与某个数组元素相同，若相同，该数组元素即为Special Element,

注意，此时，该数组元素参与计数，并打上访问过的标记，该数组元素不参与之后的相等判定了。

第一发WA,比赛中一阵紧张，算法思路都没问题，仔细排查，才发现，需要注意，数组脚标不要越界，即代码中b<=n，很关键。

AC代码如下

#include <stdio.h>
#define maxn 8010
int a[maxn],cnt[maxn],sum[maxn],vis[maxn];//cnt[]运用了桶排序的思想。
int main(){
	int t,n,i,j,b,tot;
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		for(i=1;i<=n;i++)cnt[i]=0,vis[i]=0;//vis[i]给访问过的Special Element打上标记，避免重复计算。
		for(i=1;i<=n;i++)scanf("%d",&a[i]),sum[i]=sum[i-1]+a[i],cnt[a[i]]++;//前缀和，桶排序
		tot=0;
		for(i=1;i<=n;i++)
			for(j=i+1;j<=n;j++){//注意j=i+1,因为two or more consecutive elements
				b=sum[j]-sum[i-1];//b表示数组脚标区间[i,j]之间的数组元素连续和。
				if(b<=n&&cnt[b]&&vis[b]==0)tot+=cnt[b],vis[b]=1;//关键点b<=n,这样cnt[b],vis[b]两个数组脚标就不会越界。
			}
		printf("%d\n",tot);
	}
	return 0;
}