轻拍牛头（求约数C++）

题目

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

贝茜让 N头奶牛（编号 11 到 N）坐成一个圈。

除了 11 号与 N号奶牛外，i 号奶牛与 i−1 号和 i+1号奶牛相邻，N 号奶牛与 1 号奶牛相邻。

农夫约翰用很多纸条装满了一个桶，每一张纸条中包含一个 1 到 1000000 之间的数字。

接着每一头奶牛 i 从桶中取出一张纸条，纸条上的数字用 Ai表示。

所有奶牛都选取完毕后，每头奶牛轮流走上一圈，当走到一头奶牛身旁时，如果自己手中的数字能够被该奶牛手中的数字整除，则拍打该牛的头。

牛们希望你帮助他们确定，每一头奶牛需要拍打的牛的数量。

即共有 N 个整数 A1,A2,…,AN，对于每一个数 Ai，求其他的数中有多少个是它的约数。

输入格式

第一行包含整数 N。

接下来 N 行，每行包含一个整数 Ai。

输出格式

共 N 行，第 i 行的数字为第 i头牛需要拍打的牛的数量。

数据范围

1≤N≤105,
1≤Ai≤106

输入样例：

5
2
1
2
3
4

输出样例：

2
0
2
1
3

 代码

#include

using namespace std;

const int N = 1e6 + 10;
int a[N];
int cnt[N]; //这个数出现的次数 

int main()
{
	int n;
	cin >> n;
	
	for(int i = 0; i < n; i ++) cin >> a[i], cnt[a[i]] ++;
	
	for(int i = 0; i < n; i ++)
	{
		int res = 0, t = a[i];
		
		for(int j = 1; j * j <= t; j ++) // 从1遍历到根号t 
			if(t % j == 0) //约数肯定是对称的，因此下面需要分两种情况 
			{
				if(t == j && cnt[t] > 1) res += cnt[j] - 1; // 如果这个约数是他本身，那么要去掉自己-1 
				else if(t != j && cnt[t] >= 1) res += cnt[j];// 否则全部加上 
				
				if(j * j != t) // 不会被算两次，现在算约数的另一边 
				{
					int k = t / j;
					if(k == t && cnt[k] > 1) res += cnt[k] - 1;
					else if(k != t && cnt[k] >= 1) res += cnt[k];
				}
			}
		cout << res << endl;
	} 
	
	return 0;	
}