洛谷 P2680 运输计划

题目背景
公元 2044 年,人类进入了宇宙纪元。
题目描述
L 国有 n 个星球,还有 n-1 条双向航道,每条航道建立在两个星球之间,这 n-1 条航道连通了 L 国的所有星球。
小 P 掌管一家物流公司,该公司有很多个运输计划,每个运输计划形如:有一艘物
流飞船需要从 ui 号星球沿最快的宇航路径飞行到 vi 号星球去。显然,飞船驶过一条航道 是需要时间的,对于航道 j,任意飞船驶过它所花费的时间为 tj,并且任意两艘飞船之 间不会产生任何干扰。
为了鼓励科技创新,L 国国王同意小 P 的物流公司参与 L 国的航道建设,即允许小 P 把某一条航道改造成虫洞,飞船驶过虫洞不消耗时间。
在虫洞的建设完成前小 P 的物流公司就预接了 m 个运输计划。在虫洞建设完成后, 这 m 个运输计划会同时开始,所有飞船一起出发。当这 m 个运输计划都完成时,小 P 的 物流公司的阶段性工作就完成了。
如果小 P 可以自由选择将哪一条航道改造成虫洞,试求出小 P 的物流公司完成阶段 性工作所需要的最短时间是多少?
输入输出格式
输入格式：
输入文件名为 transport.in。
第一行包括两个正整数 n、m,表示 L 国中星球的数量及小 P 公司预接的运输计划的数量,星球从 1 到 n 编号。
接下来 n-1 行描述航道的建设情况,其中第 i 行包含三个整数 ai, bi 和 ti,表示第
i 条双向航道修建在 ai 与 bi 两个星球之间,任意飞船驶过它所花费的时间为 ti。
接下来 m 行描述运输计划的情况,其中第 j 行包含两个正整数 uj 和 vj,表示第 j个 运输计划是从 uj 号星球飞往 vj 号星球。
输出格式：
输出 共1行,包含1个整数,表示小P的物流公司完成阶段性工作所需要的最短时间。
输入输出样例
输入样例#1：
6 3
1 2 3
1 6 4
3 1 7
4 3 6
3 5 5
3 6
2 5
4 5
输出样例#1：
11
说明
所有测试数据的范围和特点如下表所示

请注意常数因子带来的程序效率上的影响。

//60分 n*n 骗分 
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
const int MAXN = 100001;
int dep[MAXN],fa[MAXN],head[MAXN*2];
int eto[MAXN*2],eval[MAXN*2],enext[MAXN*2],tot;//平生第一次这样写
//大概也是余生最后一次
int ui[MAXN],vi[MAXN],n,m,ans[10005];
int ma[3001][3001],cost[MAXN];
void Add_Edge(int u,int v,int w){
    enext[++tot]=head[u];head[u]=tot;eto[tot]=v;eval[tot]=w;
}
void DFS(int u){
    dep[u]=dep[fa[u]]+1;
    for(int i=head[u];i;i=enext[i]){
        int v=eto[i];
        if(fa[u]==v) continue;
        cost[v]=eval[i];fa[v]=u;
        DFS(v);
    }
}

int calc(int i,int x,int y){
    int tot=0;
    if(dep[x]<dep[y]) swap(x,y);
    while(dep[x]>dep[y]){
        tot+=cost[x];
        ma[x][i]=1;
        x=fa[x];
    }
    while(x!=y){
        tot+=cost[x]+cost[y];
        ma[x][i]=1;ma[y][i]=1;
        x=fa[x];y=fa[y];
    }
    return tot;
}
void solve(){//m==1
    int MAX=0;long long sum=0;
    if(dep[ui[1]]<dep[vi[1]]) swap(ui[1],vi[1]);
    while(dep[ui[1]]>dep[vi[1]]){
        sum+=cost[ui[1]];
        MAX=max(MAX,cost[ui[1]]);
        ui[1]=fa[ui[1]];
    }
    while(ui[1]!=vi[1]){
        sum+=cost[ui[1]]+cost[vi[1]];
        MAX=max(MAX,cost[ui[1]]);
        MAX=max(MAX,cost[vi[1]]);
        ui[1]=fa[ui[1]];vi[1]=fa[vi[1]];
    }
    printf("%lld",sum-MAX);
}
void work(){
    int MIN=-1;
    for(int i=1;i<=m;i++) ans[i]=calc(i,ui[i],vi[i]),MIN=max(MIN,ans[i]);
    for(int i=2;i<=n;i++){
        int MAX=0;
        for(int j=1;j<=m;j++)
            MAX=max(MAX,ans[j]-cost[i]*ma[i][j]);
        if(MIN==-1||MIN>MAX) MIN=MAX;
    }
    printf("%d\n",MIN);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int u,v,w,i=1;i<=n-1;i++){
        scanf("%d%d%d",&u,&v,&w);
        Add_Edge(u,v,w);
        Add_Edge(v,u,w);
    }
    fa[1]=1;
    DFS(1);
    for(int i=1;i<=m;i++) scanf("%d%d",&ui[i],&vi[i]);
    if(m==1) solve();
    else work();
    return 0;
}

/*
整个题目的答案是由消耗时间最长的任务来决定的
二分答案的时候检查是否有路径的长度大于Ans(二分的答案)，
首先预处理出所有路径的长度，然后标记下这种路径的每一条边，扫描所有路径
如果有一条路径能让所有超过Ans的路径变得小于等于Ans，那么这个值就是合法的 
*/
#include<cstdio>
#include<algorithm>
#define MAXN 300005
using namespace std;
int n,m,tot,p[MAXN][3],Len[MAXN];
struct Node{int v,w,next;}e[MAXN<<1];
int head[MAXN],q[MAXN],dcnt,to[MAXN];
int dep[MAXN],fa[MAXN],dis[MAXN];
int siz[MAXN],top[MAXN],son[MAXN];
inline int read(){
    int x=0,f=1;char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+(c-'0');c=getchar();}
    return f*x;
}

void Add_Edge(int u,int v,int w){
    e[++tot].v=v;e[tot].w=w;
    e[tot].next=head[u];head[u]=tot;
}
void DFS1(int u){
    siz[u]=1;q[++dcnt]=u;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].v;
        if(siz[v]) continue;
        dep[v]=dep[u]+1;fa[v]=u;
        dis[v]=dis[u]+e[i].w;to[v]=e[i].w;
        DFS1(v);
        if(siz[v]>siz[son[u]]) son[u]=v;
        siz[u]+=siz[v];
    }
}
void DFS2(int u,int Top){
    top[u]=Top;
    if(son[u]) DFS2(son[u],Top);
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].v;
        if(v==fa[u]||v==son[u]) continue;
        DFS2(v,v);
    }
}
inline int LCA(int u,int v){
    while(top[u]!=top[v]){
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        u=fa[top[u]];
    }
    return dep[u]<dep[v] ? u : v;
}
int s[MAXN];
inline bool check(int mid){
    int cnt=0,lim=0;
    for(int i=1;i<=n;i++) s[i]=0;
    for(int i=1;i<=m;i++)
        if(Len[i]>mid){
            ++s[p[i][0]];++s[p[i][1]];
            s[p[i][2]]-=2;
            lim=max(Len[i]-mid,lim);
            ++cnt;
        }
    if(!cnt) return 1;
    for(int i=n;i>=2;i--) s[fa[q[i]]]+=s[q[i]];
    for(int i=2;i<=n;i++)
        if(to[i]>=lim&&s[i]==cnt) return 1;
    return 0;
}
int main(){
    n=read();m=read();
    int l=0,r=0,ans;
    for(int u,v,w,i=1;i<n;i++){
        u=read();v=read();w=read();
        Add_Edge(u,v,w);
        Add_Edge(v,u,w);
        r+=w;
    }
    DFS1(1);DFS2(1,1);
    for(int i=1;i<=m;i++){
        p[i][0]=read();p[i][1]=read();
        p[i][2]=LCA(p[i][0],p[i][1]);
        Len[i]=dis[p[i][0]]+dis[p[i][1]]-2*dis[p[i][2]];
    }
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid)) ans=mid,r=mid-1;
        else l=mid+1;
    }
    printf("%d\n",ans);
    return 0;
}