P2926 [USACO08DEC] Patting Heads S题解

题目

今天是贝茜的生日，为了庆祝自己的生日，贝茜邀你来玩一个游戏。

贝茜让N (1≤N≤) 头奶牛坐成一个圈。除了1号与N号奶牛外，i号奶牛与i−1号和i+1号奶牛相邻。N号奶牛与1号奶牛相邻。农夫约翰用很多纸条装满了一个桶，每一张包含了一个不一定是独一无二的1到的数字。

接着每一头奶牛i从桶中取出一张纸条Ai​。每头奶牛轮流走上一圈，同时拍打所有手上数字能整除在自己纸条上的数字的牛的头，然后坐回到原来的位置。牛们希望你帮助他们确定，每一头奶牛需要拍打的牛的数量。

输入输出样例

输入样例

5 
2 
1 
2 
3 
4 

输出样例

2 
0 
2 
1 
3 

解析

针对这个题目，若是从枚举约数的角度思考，会很自然地得到这样一个算法：对于每个数a[i]，枚举它的所有约数，再看这些约数在n个数中出现了几次，对它们求和即为答案。这个算法的复杂度是,比更优了，但无法通过本题。

从枚举倍数的角度思考：对于每一个数i，若它在原数组中出现了c[i]次，那么i这个数会对它的每一个倍数产生c[i]的贡献，同时和自己大小一样的数字也是自己的约数。这样就可以通过查询这样产生的贡献的总和来计算答案了。

#include<iostream>
using namespace std;
#define maxn 1000010
int n,a[maxn],c[maxn],w[maxn];
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		c[a[i]]++;
	}
	for(int i=1;i<=1000000;i++){
		for(int j=i;j<=1000000;j+=i){
			w[j]+=c[i];//i这个数对j产生c[i]的贡献 
		}
	}
	for(int i=1;i<=n;i++){
		cout<<w[a[i]]-1<<endl;
	}
	return 0;
}