Bzoj P2282 [Sdoi2011]消防___二分+前缀和+树的直径

题目大意：

有$n$个点的一棵树，任意两点都连通且有唯一一条路径。
给出一个最大允许花费$s$，要在树中选择一条边长度和不超过$s$的路径，且使其他所有点到这条路径的距离的最大值最小。
问最大值最小是多少。
$n<=300000，边长\le 1000$

分析：

树的直径：树中长度最长的一条链
我们发现需要选的路径的所有边都在直径上才是最优的，
证明：
树上任意一点$x$，离它最远的点一定是直径的某个端点，
否则必定它跟它最远的点构成的唯一路径比原先的直径长，
所以我们选择的路径如果不与直径相交那么肯定不是最优解，因为选的路径上的点到其他点的最大值并没有发生任何的改变

而为什么整条路径都在直径上会最优，
假设有一个与直径只是相交的边，那么即是只有一个公共点，我们可以发现不管怎么选边，最大值就是这个相交点到直径两个端点距离的最大值，
整合一下就是$O(n)$求直径，处理直径长度的前缀和，然后二分答案判断
二分的下界$L$就是直径上每个点去除在到两端点的路径以外的最长路径这些路径中的最大值，上界$R$就是直径长度
每次判断就看存不存在直径上 一个路径方案 使得当前二分的值合法

代码：

#include <iostream>
#include <cstring>
#include <cmath>
#include <queue>
#include <cstdio>
#include <algorithm>

#define N 300005

using namespace std;

struct Node { int To, w, nxt; }e[N*2];
int dis[N], sum[N], ls[N], maxL, maxR, cnt, lim, len, mid, ans, L, R, n;
bool vis[N], check;
 
queue <int> Q;
 
void Addedge(int u, int v, int w)
{
    e[++cnt].To = v, e[cnt].w = w, e[cnt].nxt = ls[u], ls[u] = cnt;
}

int bfs(int x)
{
    for (int i = 1; i <= n; i++) dis[i] = 0, vis[i] = 0; 
    Q.push(x); vis[x] = 1;
    int cdp, maxlen = 0;
    while (Q.size())
    {
        int u = Q.front(); Q.pop();
        for (int i = ls[u]; i; i = e[i].nxt)
            if (!vis[e[i].To])
            {
                dis[e[i].To] = dis[u] + e[i].w; 
                vis[e[i].To] = 1; 
                Q.push(e[i].To);
                if (dis[e[i].To] > maxlen) maxlen = dis[e[i].To], cdp = e[i].To;
            }
    }
    return cdp;  
}


void Work(int x, int lim1, int lim2, int tot)
{
    L = max(L, tot);
    for (int i = ls[x]; i; i = e[i].nxt)
    {
        if (e[i].To == lim1 || e[i].To == lim2) continue;
        Work(e[i].To, x, x, tot + e[i].w);
    }
}

void dfs(int x, int fa, int now, int tot)
{
    if (x == maxR) { check = 1; len = now; sum[len] = tot; R = sum[len]; return; }
    for (int i = ls[x]; i; i = e[i].nxt)
    {
        if (e[i].To == fa) continue;
        dfs(e[i].To, x, now + 1, tot + e[i].w);
        if (check) { sum[now] = tot; Work(x, fa, e[i].To, 0); return; }
    }
}

bool pan()
{
    int pos1 = 0, pos2 = 0;
    for (int i = 1; i <= len + 1; i++)
        if (sum[i] > mid) { pos1 = i - 1; break; }
    for (int i = pos1; i <= len + 1; i++)
        if (sum[i] - sum[pos1] > lim) { pos2 = i - 1; break; }
    return (sum[len] - sum[pos2]) <= mid; 
}

int main()
{
    scanf("%d %d", &n, &lim);
    for (int i = 1; i < n; i++)
    {
        int u, v, w; 
        scanf("%d %d %d", &u, &v, &w);
        Addedge(u, v, w);
        Addedge(v, u, w);
    }
    maxL = bfs(1);
    maxR = bfs(maxL);
    dfs(maxL, -1, 1, 0);
    sum[len + 1] = 0x7fffffff;
    while (L <= R)
    {
        mid = (L + R) >> 1;
        if (pan()) ans = mid, R = mid - 1; else L = mid + 1;
    }
    printf("%d\n", ans);
    return 0;
}