【gmoj3501】【luoguP2018】【7.13提高B组T1】消息传递【树上记忆化搜索】

对于100%的数据，满足N<=200000

分析

用枚举所有根然后转移的方法可以有20pts

但是发现若$a\ne b$时$a$做根和$b$做根时儿子相同，就可以拿第一次计算出来的$f_x$作为返回值。

具体就是对于每个相同的$so{n}_x$用$d[x][f{a}_x]$记录值，下次相同直接用，完成记忆化搜索。

最后用$f{a}_x$连向$x$的单向边把空间优化成$O(n)$

这个算法在洛谷稳过，但是在gmoj只能拿70pts！
要拿100pts需要用到换根树形DP+RMQ+二分优化，暂时不太会。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;

struct lwx
{
	int to,next;
}e[400001];

int n,d[400001],tj[200001],mn=0x7fffffff;
int tot,hd[200001];

void add(int x,int y)
{
	e[++tot]=(lwx){y,hd[x]};
	hd[x]=tot;
}

int dfs(int k,int fa,int p)
{
    if(p&&d[p]) return d[p];
	priority_queue<int> q;
	int ans=0;
	for(int i=hd[k];i>0;i=e[i].next)
	{
		if(e[i].to==fa) continue;
		q.push(dfs(e[i].to,k,i));
	}
	int j=0;
	while(!q.empty())
	{
		j++;
		ans=max(ans,q.top()+j);
		q.pop();
	}
	return d[p]=ans;
}

int main()
{
	freopen("news.in","r",stdin);
	freopen("news.out","w",stdout);
	cin>>n;
	for(int i=2;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		add(x,i);
		add(i,x);
	}
	for(int i=1;i<=n;i++)
	{
		tj[i]=dfs(i,0,0)+1;
		mn=min(tj[i],mn);
	}
	cout<<mn<<endl;
	for(int i=1;i<=n;i++)
	{
		if(tj[i]==mn) cout<<i<<' ';
	}
	return 0;
}