51nod-2602 树的直径
题目
给出一颗 n 个节点的树,求树上最远点对。
分析
用 d p [ i ] dp[i] dp[i] 维护以 i 为根的子树的深度, d p [ i ] = m a x ( d p [ i ] , d p [ j ] + 1 ) dp[i] = max(dp[i], dp[j]+1) dp[i]=max(dp[i],dp[j]+1)(j 为 i 的儿子)。
求出任意节点 i 子树深度之后,经过 i 的最长路径就是最深的两颗子树深度相加。
上述状态转移方程只维护了最深的,需要改变一下,现在要求出最深的两个相加。维护两个变量 m1, m2,用 if 判断即可。
最后答案是所有最长路径中最长的。
不一定是经过根节点的最长路径。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 9999991;
const int N = 1e5 + 10;
int n, x, y, r;
vector<int> son[N];
int rt[N], dp[N]; //dp[i] 代表以 i 为跟的子树深度
int DP(int rt){
dp[rt] = 0;
int m1 = 0, m2 = 0; //以 rt 为跟的最长路径就是 子树中最深的两颗深度相加
for(auto x : son[rt]){
DP(x); //通过子树深度计算当前深度之前,要把子树深度先求出来
if(dp[x] + 1 > m1){
m2 = m1;
m1 = dp[x] + 1;
}else if(dp[x] + 1 > m2){
m2 = dp[x] + 1;
}
}
dp[rt] = m1;
return m1 + m2;
}
int main()
{
scanf("%d", &n);
for (int i = 0; i < n-1; i++){
scanf("%d%d", &x, &y);
rt[y]++;
son[x].push_back(y);
}
for(int i = 1; i <= n; i++)
if(!rt[i]){
r = i; break;
}
printf("%d\n", DP(r));
return 0;
}