D48 树的直径 两次DFS+双指针 P3304 [SDOI2013] 直径

D48 树的直径 P3304 [SDOI2013] 直径_哔哩哔哩_bilibili 

P3304 [SDOI2013] 直径 - 洛谷

 

// 树的直径 两次DFS+双指针 O(n)
#include<bits/stdc++.h>
using namespace std;

#define ll long long
const int N=200005;
int n,p,l,r,pre[N],col[N];
ll mxd,cnt,d[N]; //d[u]从根走到u点的距离
vector<pair<int,int>> e[N];

void dfs(int u,int fa){
  if(d[u]>d[p]) p=u; //记录最远点
  pre[u]=fa; //记录路径
  for(auto [v,w]:e[u]){
    if(v==fa||col[v]) continue;
    d[v]=d[u]+w;
    dfs(v,u);
  }
}
int main(){
  ios::sync_with_stdio(0);
  cin>>n;
  for(int i=1,x,y,z;i<n;i++){
    cin>>x>>y>>z;
    e[x].emplace_back(y,z);
    e[y].emplace_back(x,z);
  }
  dfs(1,0);
  d[p]=0; l=p;
  dfs(p,0);
  mxd=d[p]; r=p;
  for(int i=r;i;i=pre[i]) col[i]=1; //直径染色
  
  for(int i=r;i;i=pre[i]){ //双指针收缩路径
    ll ld=d[i],rd=mxd-d[i];
    p=i,d[p]=0;
    dfs(p,pre[p]); //搜索支路
    if(d[p]==rd) r=i; //支路长度=右段长度
    if(d[p]==ld){l=i;break;} //支路长度=左段长度
  }
  for(int i=r;i!=l;i=pre[i]) cnt++;
  cout<<mxd<<"\n"<<cnt;
}