Maximal Tree Diameter [树形dp]

题意:给一颗树,删掉一条边然后连上一条边,求改变后的树的最大直径。

题解:树形dp记录每个节点向下的最长路径,次长路径,次次长路径,还有以每个节点为根时候的树的直径,通过树形dp计算出删除父节点与子节点之间边后的两个树的直径,然后加起来取最大值就是答案。

AC代码:

#include<stdio.h>
#include<vector>
#include<iostream>
using namespace std;
vector<int>vt[500005];
int dp[500005][3],ans;
int zhi[500005];
void dfs(int u,int fa)
{
	for(int i=0;i<vt[u].size();i++)
	{
		int to=vt[u][i];
		if(to==fa)continue;
		dfs(to,u);
		if(dp[to][0]+1>=dp[u][0])dp[u][2]=dp[u][1],dp[u][1]=dp[u][0],dp[u][0]=dp[to][0]+1;
		else if(dp[to][0]+1>=dp[u][1])dp[u][2]=dp[u][1],dp[u][1]=dp[to][0]+1;
		else if(dp[to][0]+1>=dp[u][2])dp[u][2]=dp[to][0]+1;
		zhi[u]=max(zhi[u],zhi[to]);
	}
	zhi[u]=max(zhi[u],dp[u][0]+dp[u][1]);
}
void dfs(int u,int fa,int sum,int zj)
{
	int b[2]={0};
	sum++;
	for(int i=0;i<vt[u].size();i++)
	{
		int to=vt[u][i];
		if(to==fa)continue;
		if(zhi[to]>=b[0])b[1]=b[0],b[0]=zhi[to];
		else if(zhi[to]>=b[1])b[1]=zhi[to];
	}
	for(int i=0;i<vt[u].size();i++)
	{
		int to=vt[u][i];
		if(to==fa)continue;
		if(dp[to][0]+1==dp[u][0])
		{
			if(b[0]==zhi[to])ans=max(ans,max(b[1],max(zj,max(sum+dp[u][1],dp[u][1]+dp[u][2])))+zhi[to]+1);
			else ans=max(ans,max(b[0],max(zj,max(sum+dp[u][1],dp[u][1]+dp[u][2])))+zhi[to]+1);
			dfs(to,u,max(sum,dp[u][1]),max(zj,dp[to][1]+dp[to][2]));
		}
		else if(dp[to][0]+1==dp[u][1])
		{
			if(b[0]==zhi[to])ans=max(ans,max(b[1],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][2])))+zhi[to]+1);
			else ans=max(ans,max(b[0],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][2])))+zhi[to]+1);
			dfs(to,u,max(sum,dp[u][0]),max(zj,dp[to][0]+dp[to][2]));
		}
		else 
		{
			if(b[0]==zhi[to])ans=max(ans,max(b[1],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][1])))+zhi[to]+1);
			else ans=max(ans,max(b[0],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][1])))+zhi[to]+1);
			dfs(to,u,max(sum,dp[u][0]),max(zj,dp[to][0]+dp[to][1]));
		}
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<n-1;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		vt[u].push_back(v);
		vt[v].push_back(u);
	}
	dfs(1,1);
	dfs(1,1,-1,0);
	printf("%d\n",ans);
} 

猜你喜欢

转载自blog.csdn.net/acterminate/article/details/79792084