树的中心

树的中心问题

  给出一棵边带权的树，求树中的一个点，使得此点到树中的其他结点的最过距离最近。

分析

  令$f[i]$表示以$i$为中心，其它点距离$i$最远的距离。
  对于一个点$i$，分两种情况：
  1.距离$i$最远的点在以$i$为根的子树中
  2.距离$i$最远的点不在以$i$为根的子树中
  初步得到状态转移方程：
  第一次从儿子节点转移给父亲：$$f[i]=max(f[son]+w[i][son])$$
  第二次从父亲节点转移给儿子：$$f[i]=max(f[i],f[fa[i]]+w[fa[i]][i])$$
  但这转移方程显然有问题，因为第二次从父亲转移给儿子时，若$f[fa[i]]$本身是由$f[i]$转移过去的，那么这样转移就是错误的。
  考虑怎样改进这种方法。
  我们发现这个方法的缺陷在于$f[fa[i]]$本身是由$f[i]$转移过去的情况，那么这时候可以用一个次优值进行转移。所以在计算的应同时记一个最优值和一个次优值。
  这就是常见的需要“换根”的树形DP。这里换根是为了消除自己影响而记录了次大值。

代码

  与上面的方法稍有差别，不过思路是一样的

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

const int Max=100005;
int n,m,s,d1[Max],d2[Max],p1[Max],p2[Max],first[Max],up[Max],ans=1e9;
struct shu{int to,next,len;}edge[Max<<1];

inline int get_int()
{
	int x=0,f=1;char c;
	for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
	if(c=='-') f=-1,c=getchar();
	for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
	return x*f;
}

inline void build(int x,int y,int z){edge[++s].next=first[x],first[x]=s,edge[s].to=y,edge[s].len=z;}

inline void init()
{
	n=get_int();
	for(int i=1;i<n;i++)
	{
	  int x=get_int(),y=get_int(),z=get_int();
	  build(x,y,z),build(y,x,z);
	}
}

inline void dfs1(int p,int fa)
{
	for(int u=first[p];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  if(to==fa) continue;
	  dfs1(to,p);
	  if(d1[to]+edge[u].len>d1[p])
	    d2[p]=d1[p],p2[p]=p1[p],d1[p]=d1[to]+edge[u].len,p1[p]=to;
	  else if(d1[to]+edge[u].len>d2[p])
	    d2[p]=d1[to]+edge[u].len,p2[p]=to;
	}
}

inline void dfs2(int p,int fa)
{
	for(int u=first[p];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  if(to==fa) continue;
	  if(p1[p]!=to) up[to]=max(d1[p],up[p])+edge[u].len;
	  else up[to]=max(d2[p],up[p])+edge[u].len;
	  dfs2(to,p);
	}
}

inline void solve()
{
	dfs1(1,0);
	dfs2(1,0);
	for(int i=1;i<=n;i++) ans=min(ans,max(up[i],d1[i]));
	cout<<ans;
}

int main()
{
	init();
	solve();
	return 0;
}