BZOJ 3629-聪明的燕姿 【数学杂题】

Description

阴天傍晚车窗外

未来有一个人在等待

向左向右向前看

爱要拐几个弯才来

我遇见谁会有怎样的对白

我等的人他在多远的未来

我听见风来自地铁和人海

我排着队拿着爱的号码牌

城市中人们总是拿着号码牌，不停寻找，不断匹配，可是谁也不知道自己等的那个人是谁。可是燕姿不一样，燕姿知道自己等的人是谁，因为燕姿数学学得好！燕姿发现了一个神奇的算法：假设自己的号码牌上写着数字S，那么自己等的人手上的号码牌数字的所有正约数之和必定等于S。

所以燕姿总是拿着号码牌在地铁和人海找数字（喂！这样真的靠谱吗）可是她忙着唱《绿光》，想拜托你写一个程序能够快速地找到所有自己等的人。

题解

首先，要知道一个很显然的东西，一个数的所有正因数的和为$\prod_{i} \sum_{j=0}^{k_i} p_i^j$

然后，只要DFS枚举每个质因子的 k <script type="math/tex" id="MathJax-Element-4">k</script>就可以了。复杂度玄学。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 200006
#define LL long long
using namespace std;
int n,P,ans0,prime[maxn];
LL ans[maxn];
bool vis[maxn];
void make_prime(){
    memset(vis,1,sizeof(vis));
    for(int i=2;i<=200000;i++){
        if(vis[i])prime[++prime[0]]=i;
        for(int j=1;j<=prime[0]&&prime[j]<=200000/i;j++){
            vis[i*prime[j]]=0;
            if(!(i%prime[j]))break;
        }
    }
}
bool isp(LL x){
    for(int i=1;prime[i]*prime[i]<=x;i++) if(!(x%prime[i]))return 0;
    return 1;
}
void dfs(int p,LL x,LL sum){
    if(sum==1){ans[++ans0]=x;return;}
    if(sum-1>=prime[p]&&isp(sum-1))ans[++ans0]=x*(sum-1);
    for(int j=p;prime[j]*prime[j]<=sum;j++){
        LL y=1,w=prime[j];
        for(;w+y<=sum;){
            y+=w;
            if(!(sum%y))dfs(j+1,w*x,sum/y);
            w*=prime[j];
        }
    }
}
int main(){
    freopen("swallow.in","r",stdin);
    freopen("swallow.out","w",stdout);
    make_prime();
    while(scanf("%d",&n)==1){
        int x=n;
        for(int i=1;i<=prime[0];i++)
         while(!(x%prime[i]))x/=prime[i];
        if(x>1)P=x;else P=-1;
        ans0=0;
        dfs(1,1,n);
        sort(ans+1,ans+1+ans0);
        printf("%d\n",ans0);
        for(int i=1;i<ans0;i++)printf("%lld ",ans[i]);
        if(ans0)printf("%d\n",ans[ans0]);
    }
    return 0;
}