Maximal Tree Diameter

最新推荐文章于 2020-09-24 22:58:54 发布

BanFcc

最新推荐文章于 2020-09-24 22:58:54 发布

阅读量376

点赞数

分类专栏： Hackerrank

本文链接：https://blog.csdn.net/BanFcc/article/details/76559570

版权

该博客探讨了一种树形结构的问题，即如何通过移除并重新连接一条边来找到树的最大直径。作者提到了需要记录最大、次大和第三大的直径，并使用动态规划(DP)方法进行解决。

摘要由CSDN通过智能技术生成

题目

一颗树，去掉一条边再连上一条边，求最大的直径。

记录最大值，次大值，次次大值？，然后DP

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=(int)5e5+10;
int dp[maxn][3],mark[maxn],mark2[maxn],zj[maxn],zjmark[maxn],zzj[maxn][2];
struct node{
	int to,nxt;
}ed[maxn<<1];
int head[maxn],cnt;
void addedge(int u,int v){
	ed[cnt].to=v;
	ed[cnt].nxt=head[u];
	head[u]=cnt++;
}
void dfs1(int u,int pre){
	for(int i=head[u];~i;i=ed[i].nxt){
		int v=ed[i].to;
		if(v==pre)continue;
		dfs1(v,u);
		if(dp[u][0]<dp[v][0]+1){
			dp[u][2]=dp[u][1];
			dp[u][1]=dp[u][0];
			dp[u][0]=dp[v][0]+1;
        mark2[u]=mark[u];
			mark[u]=v;