题目
今天是贝茜的生日,为了庆祝自己的生日,贝茜邀你来玩一个游戏。
贝茜让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;
}