AcWing 1291. 轻拍牛头（反向求约数个数）

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

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

除了 1 号与 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≤10^5,
1≤Ai≤10^6

输入样例：
5
2
1
2
3
4

输出样例：
2
0
2
1
3

思路

如果直接用暴力解法，逐个判断其他的数是不是它的约数，这样时间复杂度是O(n^2)，数据规模是10^5，会超时

假设A1是A2的约数的话，那么A2就是A1的倍数，因此可以做这么一个处理，当判定某个数Ak时，我们可以把这个数字的倍数Am全部加1，含义是，Am的约数个数又多了1。

但又考虑到可能会有很多个Ak的值是相同的，所以类似于计数排序的思想，先统计Ak出现的次数，然后再做上面的操作会方便一些

统计a[i]出现的次数：

   for(int i=0;i<n;++i)
   {
        scanf("%d",&a[i]);
        cnt[a[i]]++;
   }
   

cnt[i]!=0，即有某个Ak出现了，把Ak的倍数j加上Ak出现的次数，即j的约数个数又多了cnt[Ak]个

    for(int i=1;i<M;++i)
    {
    //i其实就是Ak，j就是i的倍数
        for(int j=i;j<M;j+=i)
        {
            s[j]+=cnt[i];
        }
    }

最后，要在所有数中求Ak约数个数，res[Ak]就是Ak约数的个数（包括Ak自己）。可以想象，假设Ak=4，它在所有数中有两个约数A1=A2=2，经过上述操作后，res[4]的值应该是3，所以Ak的约数个数是res[Ak]−1=3−1=2

总结

约数和倍数其实是一对好基友，当求约数的时间复杂度过大时，不妨从它倍数的角度来考虑解决问题

时间复杂度：

nlogn（在于双重for循环）

#include<bits/stdc++.h>

using namespace std;
const int N = 1e5+10,M = 1e6+10;
int cnt[M];
int a[N],s[M];
int n;

int main()
{
    cin>>n;
    for(int i=0;i<n;++i)
    {
        scanf("%d",&a[i]);
        cnt[a[i]]++;
    }
    
    for(int i=1;i<M;++i)
    {
        for(int j=i;j<M;j+=i)
        {
            s[j]+=cnt[i];
        }
    }
    
    for(int i=0;i<n;++i) printf("%d\n",s[a[i]]-1);//最后别忘了把自己给减去
    
    
    return 0;
}