树形DP(3/6) 1075. 数字转换

• 题意：
  如果一个数 x 的约数之和 y（不包括他本身）比他本身小，那么 x 可以变成 y，y 也可以变成 x。例如，4 可以变为 3，1 可以变为 7。限定所有数字变换在不超过 n 的正整数范围内进行，求不断进行数字变换且不出现重复数字的最多变换步数。
• 每个可行的变换可以看作两个点之间连一条边 约数和可能出现相等的 但是从1-n是不会相等的
  每次连边看作约数向编号1-n连边 即可符合树的定义 至此我们就可分析出题意所求：
  不断进行数字变换且不出现重复数字的最多变换步数 -> 树中的最长路径的长度
  树形DP解决即可

#include <bits/stdc++.h>

using namespace std;

int head[50010],nxt[100010],to[100010],tot;

void add(int u,int v) {
	to[++tot] = v,nxt[tot] = head[u],head[u] = tot;
}

int dp[10010][2],ans;
void dfs(int u,int fa) {
	for(int i = head[u];i;i = nxt[i]) {
		int v = to[i];
		if(v == fa) continue;
		dfs(v,u);
		if(dp[v][0] + 1 > dp[u][0]) {
			dp[u][1] = dp[u][0];
			dp[u][0] = dp[v][0] + 1;
		} else if(dp[v][0] + 1 > dp[u][1]) {
			dp[u][1] = dp[v][0] + 1;
		}
	}
	// cout << u << " :" << dp[u][0] + dp[u][1] << '\n';
	ans = max(dp[u][0] + dp[u][1],ans);
}

int main() {
	int n; cin >> n;
	for(int i = 1;i <= n;i ++) {
		int s = 1;
		for(int j = 2;j * j <= i;j ++) {
			if(i % j == 0) {
				s += j;
				if(j != i / j) s += i / j;
			}
		}
		if(s < i) {
			add(s,i); add(i,s);
		}
	}
	dfs(1,0);
	cout << ans << '\n';
	return 0;
}