【ACWing】1075. 数字转换

题目地址：

https://www.acwing.com/problem/content/1077/

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

输入格式：
输入一个正整数$n$。

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

数据范围：
$1\le n\le 50000$

设$s(x)={\sum }_{d|x,1\le d\le x}d$，也就是说当$y=s(x)<x$，那么$x$和$y$可以相互转化。我们考虑一下互相转化的数所构成的图有什么性质。如果$y<x$，那么有可能有另一个数$z$，也满足$y=s(z)$，这样$y$和$z$也能相互转化，但是比$x$小的且能与$x$互相转化的数是唯一的，就是$s(x)$。所以我们可以构造一个从$s(x)\to x$的有向边，这样所有转化关系形成的图就是个森林，每棵树是个有根树。问题就转化为了求树中的最长路径的长度是多少，参考https://blog.csdn.net/qq_46105170/article/details/114684812。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 50010, M = 2 * N;
int n;
int h[N], e[M], ne[M], idx;
int sum[N];
// 存每个节点是否是树根，避免重复枚举森林中的树
bool is_root[N];
int res;

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

int dfs(int u) {
    int d1 = 0, d2 = 0;
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = e[i];

        int dist = dfs(j) + 1;
        if (dist > d1) d2 = d1, d1 = dist;
        else if (dist > d2) d2 = dist;
    }

    res = max(res, d1 + d2);

    return d1;
}

int main() {
    cin >> n;

    for (int i = 1; i <= n; i++)
        for (int j = 2; j <= n / i; j++)
            sum[i * j] += i;

    memset(h, -1, sizeof h);
    memset(is_root, true, sizeof is_root);

    for (int i = 2; i <= n; i++)
        if (i > sum[i]) {
            add(sum[i], i);
            is_root[i] = false;
        }

    for (int i = 1; i <= n; i++)
        if (is_root[i])
            dfs(i);

    cout << res << endl;

    return 0;
}

时间复杂度$O(n\log n)$（主要的时间花在了两重循环上，$n({\sum }_{k=2}^n\frac{1}{k})=O(n\log n)$），空间$O(n)$。