AcWing 1075：数字转换 ← 树形DP

【题目来源】
https://www.acwing.com/problem/content/1077/

【题目描述】
如果一个数 x 的约数之和 y（不包括他本身）比他本身小，那么 x 可以变成 y，y 也可以变成 x。
例如，4 可以变为 3，1 可以变为 7。
限定所有数字变换在不超过 n 的正整数范围内进行，求不断进行数字变换且不出现重复数字的最多变换步数。

【输入格式】
输入一个正整数 n。

【输出格式】
输出不断进行数字变换且不出现重复数字的最多变换步数。

【数据范围】
1≤n≤50000

【输入样例】
7

【输出样例】
3

【样例解释】
一种方案为：4→3→1→7。

【算法代码】

/*
d1[u]:从当前结点 u 向下走的最长路径长度
d2[u]:从当前结点 u 向下走的次长路径长度
up[u]:从当前结点 u 向上走的最长路径长度
p1[u]:从当前结点 u 向下走的最长路径是从哪个结点开始的
p2[u]:从当前结点 u 向下走的次长路径是从哪个结点开始的
*/

#include <bits/stdc++.h>
using namespace std;

const int N=5e4+5;
const int M=N<<1;
int e[M],ne[M],h[N],idx;
int d1[N],d2[N],res;
int n,fsum[N];

void add(int a,int b) {
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u) {
    if(d1[u]) return;
    for(int i=h[u]; ~i; i=ne[i]) {
        int j=e[i];
        dfs(j);
        if(d1[j]+1>=d1[u]) d2[u]=d1[u],d1[u]=d1[j]+1;
        else if(d1[j]+1>d2[u]) d2[u]=d1[j]+1;
    }
    res=max(res,d1[u]+d2[u]);
}

int main() {
    memset(h,-1,sizeof h);
    cin>>n;
    for(int i=1; i<=n; i++) {
        for(int j=2; j<=n/i; j++) fsum[i*j]+=i;
    }

    for(int i=2; i<=n; i++) {
        if(fsum[i]<i) add(fsum[i],i);
    }

    for(int i=1; i<=n; i++) dfs(i);

    cout<<res<<endl;

    return 0;
}

/*
in:
7

out:
3
*/

【参考文献】
https://www.acwing.com/solution/content/65467/