[BZOJ1968][Ahoi2005]COMMON 约数研究（线性筛）

=== ===

这里放传送门

=== ===

题解

这道题的意思是求$[1,n]$内每个数约数个数的和。如果设$d(x)$表示$x$的约数个数，那么设$x=p_1^{k_1}*p_2^{k_2}*...*p_n^{k_n}$，则$d(x)=(k_1+1)*(k_2+1)*...*(k_n+1)$。这个可以用排列组合的原理来理解，因为每个质因数$p_i$都能拿出$0..k_i$个质因数来构造新的约数，共有$k_i+1$种可能性。还可以证明$d(x)$是一个积性函数，于是它可以用线性筛来求。为了求这个函数，还要记录一下最小质因子的指数$e(x)$用于最小质因子不为1的情况下的递推。具体递推过程如下：
- 对于一个数$i$，当$i$为质数的时候，$d(i)=2$，$e(i)=1$。
- 当$i$的最小质因子$p$的指数为1的时候，显然$i/p$和$p$是互质的，于是可以利用积性函数的性质，$d(i)=d(i/p)*d(p)=d(i/p)*2$。
- 当$i$的最小质因子指数不为1的时候，因为最小质因子的次数增加了1，所以只要把$d(i/p)$里面质因子$p$贡献的那一部分改一下就可以了，具体就是因为一开始$p$的指数为$e(i/p)$，为$d(i/p)$贡献的那一部分是$e(i/p)+1$，那么乘上一个$p$以后有

$$e(i)=e(i/p)+1$$ 则

$$d(i)={d(i) \over e(i)}*(e(i)+1)$$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int d[1000010],e[1000010],n,prm[1000010];
bool ext[1000010];
long long sum;
int main()
{
    scanf("%d",&n);
    d[1]=e[1]=1;sum=1;
    for (int i=2;i<=n;i++){
        if (ext[i]==false){
            d[i]=2;e[i]=1;//质数的约数个数为2，最小质因子指数为1
            prm[++prm[0]]=i;
        }
        for (int j=1;j<=prm[0];++j){
            if (i*prm[j]>n) break;
            ext[i*prm[j]]=true;//不要忘了筛去合数
            if (i%prm[j]==0){
                d[i*prm[j]]=d[i]/(e[i]+1)*(e[i]+2);//在式子里面更新最小质因子的指数
                e[i*prm[j]]=e[i]+1;
                break;
            }
            else{
                d[i*prm[j]]=d[i]*d[prm[j]];
                e[i*prm[j]]=1;
            }
        }
        sum+=d[i];//一边筛一边累加前缀和
    }
    printf("%lld\n",sum);
    return 0;
}

偏偏在最后出现的补充说明

这都不知道啥时候做的题了只是拿出来玩$Markdown$哈哈哈