NOIP 2015 && UOJ#150 运输计划

公元 2044 年，人类进入了宇宙纪元。
L 国有 n 个星球，还有 n−1 条双向航道，每条航道建立在两个星球之间，这 n−1条航道连通了 L 国的所有星球。 小 P掌管一家物流公司， 该公司有很多个运输计划，每个运输计划形如：有一艘物流飞船需要从 ui号星球沿最快的宇航路径飞行到 vi号星球去。显然，飞船驶过一条航道是需要时间的，对于航道 j，任意飞船驶过它所花费的时间为 tj，并且任意两艘飞船之间不会产生任何干扰。
为了鼓励科技创新， L 国国王同意小 P 的物流公司参与 L 国的航道建设，即允许小P把某一条航道改造成虫洞，飞船驶过虫洞不消耗时间。
在虫洞的建设完成前小 P 的物流公司就预接了 m 个运输计划。在虫洞建设完成后，这 m个运输计划会同时开始，所有飞船一起出发。当这 m个运输计划都完成时，小 P 的物流公司的阶段性工作就完成了。 如果小 P可以自由选择将哪一条航道改造成虫洞， 试求出小 P的物流公司完成阶段性工作所需要的最短时间是多少？

输入格式

第一行包括两个正整数 n,m，表示 L 国中星球的数量及小 P 公司预接的运输计划的数量，星球从 11 到 n 编号。

接下来n−1 行描述航道的建设情况，其中第 i 行包含三个整数 ai,,bi 和 ti，表示第 i 条双向航道修建在ai 与 bi两个星球之间，任意飞船驶过它所花费的时间为ti。数据保证 1≤ai,bi≤n 且0≤ti≤1000。

接下来 m 行描述运输计划的情况，其中第 j 行包含两个正整数 uj 和 vj，表示第 j 个运输计划是从 uj号星球飞往vj号星球。数据保证 1≤ui,vi≤n

输出格式

输出文件只包含一个整数，表示小 P 的物流公司完成阶段性工作所需要的最短时间。

样例一

input

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

output

11

explanation

将第 1 条航道改造成虫洞： 则三个计划耗时分别为：11,12,11，故需要花费的时间为 12。

将第 2 条航道改造成虫洞： 则三个计划耗时分别为：7,15,11，故需要花费的时间为 15。

将第 3 条航道改造成虫洞： 则三个计划耗时分别为：4,8,11，故需要花费的时间为 11。

将第 4 条航道改造成虫洞： 则三个计划耗时分别为：11,15,5，故需要花费的时间为 15。

将第 5 条航道改造成虫洞： 则三个计划耗时分别为：11,10,6，故需要花费的时间为 11。

故将第 3 条或第 5 条航道改造成虫洞均可使得完成阶段性工作的耗时最短，需要花费的时间为 11。

限制与约定

第二次写这个题，上次写着写着就吐了，这次又想起来这道题，写了一下。
吐槽一下：写完之后去codevs交，怎么优化最后一个点都是TLE，本机跑……也得要1.5s，然后想到UOJ【滑稽。去UOJ上去交，woc最后一个点900ms过了……UOJ真是太强啦！

首先看到这个题是最大时间最短，可以想到二分。而在树上的话，显然要求lca。
那么每次二分完一个答案要怎么验证呢？枚举哪个计划的耗时超过了答案，那么我们就要在这条路径上设置虫洞，那具体在哪条边上设置虫洞呢？如果有k个计划的耗时超过了答案ans，那我们找一条被覆盖了k次的边进行修改肯定是最优的，这就用到了树上前缀和。
如果一个计划的耗时超过了ans，我们在u，v处+1，在lca（u，v）处-2。最后更新树上的前缀和，找到那条边。如果最大耗时减去那条边的花费小于ans，返回true：否则返回false。
记得每次检验答案的时候，要清空数组！！！

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int SZ = 300010;
const int INF = 1e9;

struct Edge
{
    int f, t, d;
}es[SZ << 1];
struct Plan
{
    int u, v, t, lca;
}p[SZ];
int first[SZ << 1], nxt[SZ << 1], tot = 1;
int n, m, depth[SZ], fa[SZ], dis[SZ], jump[SZ][25];
int recover[SZ], in[SZ], cnt, cost[SZ];
bool vis[SZ];

void read(int &num)
{
    int f = 1;
    num = 0;
    char c = getchar();
    while(c < '0' || '9' < c)
    {
        if(c == '-') f = -1;
        c = getchar();
    }
    while('0' <= c && c <= '9')
    {
        num = num * 10 + (c -'0');
        c = getchar();
    }
    num *= f;
}

void build(int f, int t, int d)
{
    es[++tot] = (Edge){f, t, d};
    nxt[tot] = first[f];
    first[f] = tot;
}

void dfs(int u)
{
    for(int i = 1; i <= 20; i++)
        jump[u][i] = jump[jump[u][i - 1]][i - 1];
    vis[u] = 1;
    in[++cnt] = u;
    for(int i = first[u]; i; i = nxt[i])
    {
        int v = es[i].t;
        if(vis[v]) continue;
        fa[v] = jump[v][0] = u;
        depth[v] = depth[u] + 1;
        dis[v] = dis[u] + es[i].d;
        cost[v] = es[i].d;
        dfs(v);
    }
}

int get_lca(int u, int v)
{
    if(depth[u] < depth[v]) swap(u, v);
    int t = depth[u] - depth[v];
    for(int i = 0; i <= 20; i++)
        if((1 << i) & t) u = jump[u][i];
    for(int i = 20; i >= 0; i--)
        if(jump[u][i] != jump[v][i])
        {
            u = jump[u][i];
            v = jump[v][i];
        }
    if(u == v) return v;
    return jump[u][0];  
}

bool check(int mid)
{
    int tot = 0, maxx = 0;
    for(int i = 1; i <= n; i++) recover[i] = 0;
    for(int i = 1; i <= m; i++)
        if(p[i].t > mid)
        {
            tot++;
            maxx = max(maxx, p[i].t);
            recover[p[i].u]++;
            recover[p[i].v]++;
            recover[p[i].lca] -= 2;
        }
    for(int i = cnt; i >= 1; i--)
        recover[jump[in[i]][0]] += recover[in[i]];
    for(int i = 1; i <= n; i++)
        if(recover[i] == tot)
            if(maxx - cost[i] <= mid) return true;
    return false;
}

int main()
{
    int __size__ = 1024 << 15;
    char * __p__ = (char *) malloc(__size__) + __size__;
    __asm__("movl %0, %%esp\n" :: "r"(__p__));
    freopen("transport.in","r",stdin);
    freopen("transport.out","w",stdout);
    read(n); read(m);   
    int u, v, w;
    for(int i = 1; i < n; i++)
    {
        read(u); read(v); read(w);
        build(u, v, w);
        build(v, u, w);
    }   
    dfs(1);
    for(int i = 1; i <= m; i++)
    {
        read(p[i].u);
        read(p[i].v);
        int lca = get_lca(p[i].u, p[i].v);
        p[i].lca = lca;
        p[i].t = dis[p[i].u] + dis[p[i].v] - 2 * dis[lca];
    }
    int l = 0, r = INF, mid, ans = -1;
    while(l <= r)
    {
        mid = (l + r) >> 1;
        if(check(mid)) r = mid - 1;
        else l = mid + 1;
    }
    printf("%d", l);
    fclose(stdin);
    fclose(stdout);
    return 0;
}