BZOJ_P4326[NOIP]2015 运输计划(LCA+Tarjan+二分)

NOIP2015 运输计划

Time Limit: 30 Sec Memory Limit: 128 MB
Submit: 248 Solved: 166
[Submit][Status][Discuss]
Description

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

Input

第一行包括两个正整数 n,m，表示 L 国中星球的数量及小 P 公司预接的运输计划的数量，星球从 1 到 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

Output

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

Sample Input
6 3
1 2 3
1 6 4
3 1 7
4 3 6
3 5 5
3 6
2 5
4 5
Sample Output
11

HINT
将第 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。

Source

一遍LCA，存LCA和dis，二分答案，dfs找最长路；
这个算法在BZOJ上能A，其他评测网站就呵呵了，把vector改成邻接表表示应该会快一些

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#include<iostream>
using namespace std;
#define N 300005
struct Edge{int fr,to,d;};struct qEdge{int fr,to,lca,dis;};
vector<Edge> edge;vector<int> g[N];vector<qEdge> qedge;vector<int> qs[N];
int d[N],v[N];int f[N],p[N];int n,m,k,l,r,mid,t,maxn,maxl;bool b[N];

void Add_Edge(int fr,int to,int dis){
    edge.push_back((Edge){fr,to,dis});edge.push_back((Edge){to,fr,dis});
    k=edge.size();g[fr].push_back(k-2);g[to].push_back(k-1);
}
void Add_qEdge(int fr,int to){
    qedge.push_back(qEdge{fr,to,0});
    k=qedge.size();qs[fr].push_back(k-1);qs[to].push_back(k-1);
}
int in(){
    int x=0;char ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
int Find(int x){return f[x]==x?x:f[x]=Find(f[x]);}
long long abs(long long x){return x<0?-x:x;}
void dfs(int u,int f,long long w){
    d[u]=w;int l=g[u].size();
    for(int i=0;i<l;i++){
        Edge &e=edge[g[u][i]];
        if(e.to==f) continue;
        dfs(e.to,u,w+e.d);
    }
}
void dfs2(int x,int f){
    p[x]=f;int l=g[x].size();
    for(int i=0;i<l;i++){
        Edge &e=edge[g[x][i]];
        if(e.to!=f) dfs2(e.to,x);
    }
}
void Tarjan_Lca(int u){
    f[u]=u;b[u]=true;int to,l;l=g[u].size();
    for(int i=0;i<l;i++){
        Edge &e=edge[g[u][i]];
        if(!b[e.to]){
            Tarjan_Lca(e.to);f[e.to]=u;
        }
    }
    l=qs[u].size();
    for(int i=0;i<l;i++){
        qEdge &e=qedge[qs[u][i]];
        to=(u==e.to?e.fr:e.to);
        if(b[to]) e.lca=Find(to),e.dis=abs(d[to]+d[u]-2*d[e.lca]);
    }
}
int work(int x,int f){
    int sum=v[x];int l=g[x].size();
    for(int i=0;i<l;i++)
        if(edge[g[x][i]].to!=f) sum+=work(edge[g[x][i]].to,x);
    if(sum==t) maxn=max(maxn,d[x]-d[p[x]]);
    return sum;
}
int cmp(qEdge x,qEdge y){return x.dis>y.dis;}
void init(){
    n=in(),m=in();int x,y,z;
    for(int i=1;i<n;i++){
        x=in(),y=in(),z=in();
        Add_Edge(x,y,z);
    }
    for(int i=0;i<m;i++){
        x=in(),y=in();
        Add_qEdge(x,y);
    }
    for(int i=1;i<=n;i++) f[i]=i;
    memset(d,0,sizeof(d));memset(b,0,sizeof(b));
    dfs(1,0,0);Tarjan_Lca(1);dfs2(1,1);
    sort(qedge.begin(),qedge.end(),cmp);
    maxl=qedge[0].dis;l=0,r=maxl;
/*  for(int i=0;i<m;i++){
        qEdge &e=qedge[i];
        printf("%d %d %d %d\n",e.fr,e.to,e.lca,e.dis);
    }*/
    while(l<r){
        t=maxn=0;mid=(l+r)>>1;
        memset(v,0,sizeof(v));
        for(int i=0;i<m;i++){
            qEdge &e=qedge[i];
            if(e.dis>mid)
                t++,v[e.fr]++,v[e.to]++,v[e.lca]-=2;
            else break;
        }
        work(1,0);
        if(maxl-maxn<=mid) r=mid;
        else l=mid+1;
    }
    printf("%d\n",l);
}
int main(){
    init();return 0;
}