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);
}