bzoj 2282 [Sdoi2011]消防（树的直径，二分）

Description

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

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

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

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

Input

输入包含n行：
第1行，两个正整数n和s，中间用一个空格隔开。其中n为城市的个数，s为路径长度的上界。设结点编号以此为1，2，……，n。
从第2行到第n行，每行给出3个用空格隔开的正整数，依次表示每一条边的两个端点编号和长度。例如，“2 4 7”表示连接结点2与4的边的长度为7。

Output

输出包含一个非负整数，即所有城市到选择的路径的最大值，当然这个最大值必须是所有方案中最小的。

Sample Input

【样例输入1】

5 2
1 2 5
2 3 2
2 4 4
2 5 3

【样例输出1】

【样例输入2】

8 6
1 3 2
2 3 2
3 4 6
4 5 3
4 6 4
4 7 2
7 8 3

【样例输出2】

Sample Output

HINT

【数据规模和约定】

对于20%的数据，n<=300。

对于50%的数据，n<=3000。

对于100%的数据，n<=300000，边长小等于1000。

【思路】

树的直径，二分法

可以知道题目要找的路径一定是树的直径的一段。对于这段在树上的路径，到达它的最大值有两种情况:一种是到达路径端点，这时候最大值为树的直径的两端取较大；另一种是和路径上除端点外的节点相连。

求出其他节点到直径的最大值mx，则答案一定不小于该最大值，并以之为下界二分直径端点到路径端点的距离，取最小。

【代码】

 #include<queue>

 #include<cstdio>

 #include<vector>

 #include<cstring>

 #include<iostream>

 #include<algorithm>

 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)

 using namespace std;

 const int N = *1e5+;

 const int INF = 1e9;

 struct Edge{ int v,w; };

 vector<Edge> g[N];

 int n,S,que[N],qz,dis[N],mark[N],fa[N];

 void read(int& x) {

     char c=getchar(); int f=; x=;

     while(!isdigit(c)) {if(c=='-')f=-; c=getchar();}

     while(isdigit(c)) x=x*+c-'',c=getchar();

     x*=f;

 }

 queue<int> q;

 void bfs(int u) {

     q.push(u); fa[u]=-;

     FOR(i,,n) dis[i]=INF; dis[u]=;

     while(!q.empty()) {

         int u=q.front(); q.pop();

         for(int i=;i<g[u].size();i++) {

             int v=g[u][i].v;

             if(v!=fa[u]) {

                 fa[v]=u; q.push(v);

                 if(mark[v]) dis[v]=dis[u];

                 else dis[v]=min(dis[v],dis[u]+g[u][i].w);

             }

         }

     }

 }

 bool can(int ML) {

     int l=,r=qz;

     while(l<=qz && que[]-que[l+]<=ML) l++;

     while(r && que[r-]-que[qz]<=ML) r--;

     return que[l]-que[r]<=S;

 }

 int main() {

     read(n),read(S);

     int u,v,w,x=,y=,dist,L=,R=,M;

     FOR(i,,n-) {

         read(u),read(v),read(w);

         g[u].push_back((Edge){v,w});

         g[v].push_back((Edge){u,w});

     }

     bfs(); FOR(i,,n) if(dis[i]>dis[x]) x=i;

     bfs(x); FOR(i,,n) if(dis[i]>dis[y]) y=i;

     int t=y; R=dis[y]-dis[x];

     while(t!=x) {

         que[++qz]=dis[t]; mark[t]=;

         t=fa[t];

     }

     que[++qz]=dis[t];

     bfs(x);

     FOR(i,,n) L=max(L,dis[i]);

     while(L<R) {

         M=(L+R)>>;

         if(can(M)) R=M; else L=M+;

     }

     printf("%d",L);

     return ;

 }