树的最长路径(树的直径/最远点对)定义:对于一棵n结点无根树的一条最长路径的长度,也可以说是,两点间的最远距离。(概念很简单的。)
解决方案:
以一个结点为根节点(无根树-> 有根树),求出每个节点开始的最长边与次长边。定义数组first[],second[],来分别记录。
res= max(res, first(i)+second(i))(1<=i<=n)
状态转移:
intfirst[N], second[N];//first 存储到达最远结点的长度,second记录次长度
intres = 0;
void dp(int u,intpre)
{
int k= g[u].size();
for(inti =0; i< k; i++)
{
if(g[u][i]==pre)continue;//子结点==父结点编号边重复不需要重复扫
dp(g[u][i],u);
/**当子结点的最长边+1 (加上到父结点的距离1)大于父结点的最长边更新父结点的最长边和次长边**/
if(first[g[u][i]]+1>=first[u])second[u]=first[u],first[u]=first[g[u][i]]+1;
else if(first[g[u][i]]+1>second[u])second[u]=first[g[u][i]]+1;
}
/**在以当前结点为根节点的子树中,最远两个结点的距离为first+second**/
res = max(res, first[u]+second[u]);
return ;
}
来让我们来看一下代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 105;
vector<int>g[N];
int first[N], second[N];//first 存储到达最远结点的长度,second记录次长度
int res = 0;
void dp(int u, int pre)
{
int k = g[u].size();
for(int i = 0; i < k; i++)
{
if(g[u][i]==pre)continue;///子结点 == 父结点编号 边重复不需要重复扫
dp(g[u][i],u);
/**当子结点的最长边+1 (加上到父结点的距离1)大于父结点的最长边 更新父结点的最长边和次长边**/
if(first[g[u][i]]+1>=first[u])second[u]=first[u],first[u]=first[g[u][i]]+1;
else if(first[g[u][i]]+1>second[u])second[u]=first[g[u][i]]+1;
}
/**在以当前结点为根节点的子树中,最远两个结点的距离为first+second**/
res = max(res, first[u]+second[u]);
return ;
}
int main()
{
int n,u,v;
scanf("%d",&n);
for(int i=1; i<n; i++)
{
scanf("%d %d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
dp(1,0);
printf("%d\n",res);
}