『算法-ACM 竞赛-数学-数论』因子和线性筛 （模板)

ACM 常用模板合集

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define N 500022
int prime[N],cnt;
bool vis[N];
int num[N],e[N];
void init()
{
    for(int i=2;i<=N;++i)
    {
        if(!vis[i])
        {
            prime[++cnt]=i;
            num[i]=i+1;
            e[i]=1;
        }
        for(int j=1;j<=cnt;++j)
        {
            if(prime[j]*i>N) break;
            vis[prime[j]*i]=true;
            if(i%prime[j]==0)
            {
                num[i*prime[j]]=num[i]*prime[j]+e[i];
                e[i*prime[j]]=e[i];
                break;
            }
            num[i*prime[j]]=num[i]*(prime[j]+1);
            e[i*prime[j]]=num[i];
        }
    }
}
int main()
{
	 int T;
	 init();
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        printf("%lld\n",num[n]);
    }
}