思想:首先这是一个找图最长路问题,我们假设最远路为(s,t),那么2次DFS即可找到(s,t),第一次DFS从任意一点k出发找k能到达的最远路,比如说为(k,T),那么根据证明(证明)
T一定为最长路的(s,t)的其中一个端点s或t,假设第一次找到的端点为s(即T = s),第二次从s出发,那么找到最远路的端点一定为t;(证明)
#include<iostream>
#include<vector>
using namespace std;
struct node{
int id;
int val;
node(int tid,int tval){
id = tid;
val = tval;
}
};
vector<node>vc[10010]; //存图
int vis[10010] = {0}; //是否访问
int duan;
int sum; //记录已经走了多远
int tmax = 0; //目前找到的最远距离
int des; //最远路的目的端点
void dfs(int pre,int cur,int val){ //cur为当前所到达的节点,pre为当前节点的前一节点,
//val为两节点间距离
vis[cur] = 1;
if(pre != -1){ //当pre = -1时,标识cur为起点,没有前节点
sum += val;
if(tmax < sum){
tmax = sum;
des = cur;
}
}
for(int i = 0;i < vc[cur].size();i++){
if(vis[vc[cur][i].id] != 1){
dfs(cur,vc[cur][i].id,vc[cur][i].val);
}
}
vis[cur] = 0;
sum -= val;
}
int main(){
int n;
cin>>n;
for(int i = 0;i < n - 1;i++){
int a,b,c;
cin>>a>>b>>c;
vc[a].push_back(node(b,c));
vc[b].push_back(node(a,c));
}
dfs(-1,1,0); //这里我们选择从1出发
dfs(-1,des,0); //此时的des为从1出发到达的最远点,这里我们紧接着从des出发,
//这样就可以找到最长路
cout<<tmax;
return 0;
}