1075. 数字转换 (埃氏筛+树形dp)

题目
题意: 如果一个数 x 的约数之和 y（不包括他本身）比他本身小，那么 x 可以变成 y，y 也可以变成 x。对从1到n的所有数进行建树操作，求不出现重复数字的前提下最多能够变换多少次。
思路: 先预处理一下每个数的因数之和，拿埃氏筛筛一下。之后这个问题就变成了以1为根，求树的最长路径，按1072题的树形dp跑一下即可。
时间复杂度: O(nloglogn + n)
代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 5e4+10;
#define mem(a,x) memset(a,x,sizeof(a))
int h[N],e[N],ne[N],idx = 0;
void add(int a,int b)
{
    e[idx] = b,ne[idx] = h[a],h[a] = idx++;
}
int sum[N];
int ans = 0;
int n,m,k,T;
void get(int n) //埃式筛预处理约数之和
{
	for(int i=1;i<=n;++i)
	{
		for(int j=2;j<=n/i;++j) //n+n/2+n/3
		{
			sum[i*j] += i;
		}
	}
}
int dfs(int cur)
{
	int d1 = 0,d2 = 0;
	for(int i=h[cur];~i;i=ne[i])
	{
		int j = e[i];
		int d = dfs(j) + 1;
		if(d>=d1) d2=d1,d1=d;
		else if(d>d2) d2=d;
	}
	ans = max(ans,d1+d2);
	return d1;
}
void solve()
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n; mem(h,-1); idx = 0;
	get(n);
	for(int i=2;i<=n;++i)
	{
		if(sum[i] < i) add(sum[i],i);
	}
	dfs(1);
	cout<<ans;
}
signed main(void)
{
	solve();
	return 0;
}