约数——聪明的燕姿（约数之和计算）

传送门：1296. 聪明的燕姿 - AcWing题库

思路：

最简单的思路就是在1~n-1的范围内一个一个枚举求约数和，但数据范围太大，肯定会超时。

设一个数为n=p^a*p1^a1*...pk^ak;

则该数的约数之和就是(1+p+p^2+..+p^a)*(1+p1+p1^2+...+p1^a1)*...

在此基础上可以枚举质因数和该质因数的个数，假如当前有一个质因数p满足n%(1+p+p^2)==0,则答案上面就要累乘p^2,当n==1时ans就是其中一个合法方案.

为此，要先预处理出1~（√2e9）的所有质数。

重要优化：

枚举质因数时特判n-1是否为质数，且n大于当前枚举的质因数，若成立说明ans*（n-1）也是一个合法方案，下面在枚举时也只需要枚举到√n，

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<unordered_map>
using namespace std;
typedef pair<int,int>PII;
typedef long long ll;
const int N=1e5+10;
int primes[N],cnt;
PII factor[2000];
int d[N],cntd;
int cntf;
bool st[N];
bool is_primes(int n)
{
    if(n<50000)return !st[n];

    for(int i=1;i<=cnt&&primes[i]<=n/primes[i];i++)
        if(n%primes[i]==0)
        return false;
    return true;
}
void get_primes(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(!st[i])
            primes[++cnt]=i;

        for(int j=1;primes[j]<=n/i;j++)
        {
            int t=primes[j]*i;
            st[t]=true;
            if(i%primes[j]==0)
                break;
        }
    }
}
void dfs(int n,int u,int ans)
{
    if(n==1)
    {
        d[++cntd]=ans;
        return ;
    }
    if(n>primes[u]&&is_primes(n-1)) d[++cntd]=ans*(n-1);
    
    for(int i=u;primes[i]<=n/primes[i];i++)
    {
        int p=primes[i];
        for(int x=1+p,y=p;x<=n;y*=p,x+=y)
            if(n%x==0)
            dfs(n/x,i+1,ans*y);
    }
}
int main()
{
    get_primes(50000);
    st[1]=1;
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        cntd=0;
        dfs(n,1,1);
        printf("%d\n",cntd);
        if(cntd)
        {
            sort(d+1,d+1+cntd);
        for(int i=1;i<=cntd;i++)
            printf("%d ",d[i]);
        puts("");
        }
    }
    return 0;
}