☆P2491 [SDOI2011]消防-树的直径，LCA，dfs，倍增

某个国家有n个城市，这n个城市中任意两个都连通且有唯一一条路径，每条连通两个城市的道路的长度为zi(zi<=1000)。

这个国家的人对火焰有超越宇宙的热情，所以这个国家最兴旺的行业是消防业。由于政府对国民的热情忍无可忍（大量的消防经费开销）可是却又无可奈何（总统竞选的国民支持率），所以只能想尽方法提高消防能力。

现在这个国家的经费足以在一条边长度和不超过s的路径（两端都是城市）上建立消防枢纽，为了尽量提高枢纽的利用率，要求其他所有城市到这条路径的距离的最大值最小。

你受命监管这个项目，你当然需要知道应该把枢纽建立在什么位置上

https://www.luogu.org/problemnew/show/P2491

这道题先求出树的直径，然后我们从底部开始往上枚举，很显然的一个贪心：固定了一个端点之后，另一端点越远越好。所以我们直接枚举端点，找到它的另一端，左右指针扫直径。
这时候我通过LCAO(1)来计算距离，就可以把此情况的最大距离用O(n)处理得到。

注意这题的ans要设一个无穷大的值，0x3f3f3f能过，0x3f3f就过不了了......

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=300001;
const int M=600001;
int n,s,cnt,maxn,root,t,l,r,sum,ans=0x3f3f3f;
struct node
{
    int next,to,v;
}e[M];
int head[N],dis[N],dep[N],f[N][21],lg[N];
bool vis[N];
inline void add(int x,int y,int w)
{
    e[++cnt].next=head[x];
    e[cnt].to=y;
    e[cnt].v=w;
    head[x]=cnt;
}
void dfs1(int u,int fa)
{
    for(int i=head[u];i;i=e[i].next){
        int to=e[i].to;
        if(to==fa) continue;
        dis[to]=dis[u]+e[i].v;
        dfs1(to,u);
    }
}
void dfs2(int u,int fa)
{
    for(int i=head[u];i;i=e[i].next){
        int to=e[i].to;
        if(to==fa) continue;
        dis[to]=dis[u]+e[i].v;
        f[to][0]=u;
        dep[to]=dep[u]+1;
        dfs2(to,u);
    }
}
int lca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    while(dep[x]>dep[y]){
		x=f[x][lg[dep[x]-dep[y]]-1];
	}
    if(x==y) return x;
    for(int i=lg[dep[x]];i>=0;i--)
    {
        if(f[x][i]!=f[y][i])
            x=f[x][i],y=f[y][i];
    }
    return f[x][0];
}
void init()
{
    for(int i=1;i<=9;i++){
        for(int j=1;j<=n;j++){
            f[j][i]=f[f[j][i-1]][i-1];
        }
    }
}
int main()
{
    scanf("%d%d",&n,&s);
    for(int i=1;i<=n;i++){
        lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    }
    for(int i=1;i<=n-1;i++){
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);add(y,x,w);
    }
    dfs1(1,0);
    for(int i=1;i<=n;i++){
        if(!root||dis[i]>dis[root]) root=i;
    }
    memset(dis,0,sizeof(dis));
    dep[root]=1;
    dfs2(root,0);
    for(int i=1;i<=n;i++){
        if(!t||dis[i]>dis[t]) t=i;
    }
    init();
    l=t;r=t;vis[t]=1;
    while(l!=0){
        sum=0;
        if(dis[r]-dis[l]>s){
            vis[r]=0;
            r=f[r][0];
            vis[r]=1;	
        }
        else{
            while(dis[r]-dis[l]<=s&&l!=0){
                l=f[l][0];
                vis[l]=1;
            }
            int rlca;
            for(int i=1;i<=n;i++){
                if(vis[i]) continue;
                int lca1=lca(l,i),lca2=lca(r,i);
                if(dep[lca1]>dep[lca2]) rlca=lca1;
                else rlca=lca2;
                if(dep[rlca]<dep[l]){
                    sum=max(sum,dis[l]+dis[i]-2*dis[rlca]);
                }
                else{
                    sum=max(sum,dis[i]-dis[rlca]);
                }
            }
            ans=min(ans,sum);
            l=f[l][0];vis[l]=1;
        }
    }
    printf("%d",ans);
    return 0;
}