D49 樹的直徑 P2491 [SDOI2011] 消防

影片連結：

P2491 [SDOI2011] 消防 - 洛谷 | 電腦科學教育新生態 (luogu.com.cn)

// 兩次DFS+雙指標 O(n)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=300010;
int idx,head[N];
struct edge{int to,w,ne;}e[N<<1];
void add(int u,int v,int w){
  e[++idx]={v,w,head[u]},head[u]=idx;
}
int n,s,p,r,d[N],fa[N],col[N];
int ans=2e9;

void dfs(int x,int ff){
  fa[x]=ff; //記錄路徑
  if(d[x]>d[p]) p=x; //更新端點
  for(int i=head[x];i;i=e[i].ne){
    int y=e[i].to,w=e[i].w;
    if(y==ff||col[y]) continue;
    d[y]=d[x]+w;
    dfs(y,x);
  }
}
int main(){
  scanf("%d%d",&n,&s);
  for(int i=1,u,v,w;i<n;i++){
    scanf("%d%d%d",&u,&v,&w);
    add(u,v,w);add(v,u,w);
  }
  dfs(1,0);
  d[p]=0;
  dfs(p,0); r=p; //直徑右端點
  
  for(int i=r,j=r;i;i=fa[i]){ //雙指標
    while(d[j]-d[i]>s) j=fa[j];
    ans=min(ans,max(d[i],d[r]-d[j]));
  } //直徑上的答案
  for(int i=r;i;i=fa[i])col[i]=1; //直徑染色
  for(int i=r;i;i=fa[i]){
    p=i;d[i]=0;
    dfs(i,fa[i]);
  } //直徑外的答案
  for(int i=1;i<=n;i++)ans=max(ans,d[i]);
  printf("%d\n",ans);
}