树形dp的实现方法

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

inline int read()
{
	int x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9')
	{
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9')
	{
		x=(x<<3)+(x<<1)+(ch^48);
		ch=getchar();
	}
	return x*f;
}

const int maxn=1e5+5;
int n,k,cnt,d,rt;
int head[maxn],to[maxn<<1],pre[maxn<<1];

void link(int a,int b)
{
	to[++cnt]=b;
	pre[cnt]=head[a];
	head[a]=cnt;
}

int f[maxn],g[maxn];
int dfs(int x,int fa)
{
	for(int i=head[x];i;i=pre[i])
	{
		int y=to[i];
		if(y!=fa)
		{
			int tmp=dfs(y,x);
			if(tmp>f[x]) g[x]=f[x],f[x]=tmp;
			else if(tmp>g[x]) g[x]=tmp;
		}
	}
	if(g[x]+f[x]>d) d=g[x]+f[x];
	return f[x]+1;
}

int dfs1(int x,int fa)
{
	for(int i=head[x];i;i=pre[i])
	{
		int y=to[i];
		if(y!=fa)
		{
			int tmp=dfs1(y,x);
			if(tmp>f[x]) g[x]=f[x],f[x]=tmp;
			else if(tmp>g[x]) g[x]=tmp;
		}
	}
	return f[x]+1; 
}

bool cmp(int a,int b)
{
	return a>b;
}

int main()
{
	n=read(),k=read();
	for(int i=1;i<n;i++)
	{
		int a=read(),b=read();
		link(a,b);link(b,a);
	}
	
	dfs(1,0);
	for(int i=1;i<=n;i++) f[i]=g[i]=0;
	
	int del=INT_MAX;
	for(int i=1;i<=n;i++)
	{
		if(f[i]+g[i]==d)
		{
			if(f[i]-g[i]<del) rt=i;
		}
	}
	
	dfs1(rt,0);
	sort(f+1,f+1+n,cmp);
	
	printf("%d",f[k+1]+1);
	
	return 0;
}