【洛谷P1967】【NOIP2013】货车运输

题目描述

A 国有 n 座城市，编号从 1 到 n，城市之间有 m 条双向道路。每一条道路对车辆都有重量限制，简称限重。现在有 q 辆货车在运输货物， 司机们想知道每辆车在不超过车辆限重的情况下，最多能运多重的货物。

输入输出格式

输入格式：

输入文件名为 truck.in。

输入文件第一行有两个用一个空格隔开的整数 n，m，表示 A 国有 n 座城市和 m 条道

路。 接下来 m 行每行 3 个整数 x、 y、 z，每两个整数之间用一个空格隔开，表示从 x 号城市到 y 号城市有一条限重为 z 的道路。意：x 不等于 y，两座城市之间可能有多条道路。

接下来一行有一个整数 q，表示有 q 辆货车需要运货。

接下来 q 行，每行两个整数 x、y，之间用一个空格隔开，表示一辆货车需要从 x 城市运输货物到 y 城市，注意：x 不等于 y。

输出格式：

输出文件名为 truck.out。

输出共有 q 行，每行一个整数，表示对于每一辆货车，它的最大载重是多少。如果货

车不能到达目的地，输出-1。

输入输出样例

输入样例#1：
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3
输出样例#1：
3
-1
3

说明

对于 30%的数据，0 < n < 1,000，0 < m < 10,000，0 < q< 1,000； 对于 60%的数据，0 < n < 1,000，0 < m < 50,000，0 < q< 1,000； 对于 100%的数据，0 < n < 10,000，0 < m < 50,000，0 < q< 30,000，0 ≤ z ≤ 100,000。

题解

又是一道经典的不能在经典的题==
30分思路：暴力 复杂度O（nm+qm）；
60分思路：并查集+最大生成树+暴力 复杂度O(mlogm+qm)；
100分思路：并查集+最大生成树+倍增LCA 复杂度O(mlogm+qlogm);

首先把给出的所有边从大到小排序（雾）然后用kruskal并查集做最大生成树
然后构造倍增LCA
此处要用到加权LCA:一个数组维护每个点的第个祖先，另一个数组维护每个点到其第个祖先权值最小的边
转移方程如下：

然后就和LCA一样查询就可以辣！

My Code

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
using namespace std;
struct wn{
    int x,y,w;
}e[50005],shu[20005];
int fa[10005],he[10005],vis[10005],dep[10005],f[10005][20],weight[10005][20],n,m,q,po=0;
int cmp(wn x,wn y){
    return x.w>y.w;
}
int find(int x){
    if(fa[x]==x)return x;
    fa[x]=find(fa[x]);
    return fa[x];
}
void addedge(int x,int y,int w){
    po++;
    shu[po].x=y;
    shu[po].y=w;
    shu[po].w=he[x];
    he[x]=po;
    po++;
    shu[po].x=x;
    shu[po].y=w;
    shu[po].w=he[y];
    he[y]=po;
}
void dfs(int x,int faq,int w){
    dep[x]=dep[faq]+1;
    f[x][0]=faq;
    weight[x][0]=w;
    vis[x]=1;   
    for(int i=he[x];i!=0;i=shu[i].w){
        if(!vis[shu[i].x]){
            dfs(shu[i].x,x,shu[i].y);
        }
    }
}
void csh(){
    for(int i=1;i<=18;i++){
        for(int j=1;j<=n;j++){
            f[j][i]=f[f[j][i-1]][i-1];
            weight[j][i]=min(weight[j][i-1],weight[f[j][i-1]][i-1]);
        }
    }
}
void hb(int x,int y,int w){
    int xx=find(x),yy=find(y);
    if(xx!=yy){
        fa[xx]=yy;
        addedge(x,y,w);
    }
}
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    int minn=20010412*10;
    for(int i=18;i>=0;i--){
        if(dep[f[x][i]]>=dep[y]){
            minn=min(weight[x][i],minn);
            x=f[x][i];  
        }
    }
    if(x==y)return minn;
    for(int i=18;i>=0;i--){
        if(f[x][i]!=f[y][i]){
            minn=min(minn,weight[x][i]);
            minn=min(minn,weight[y][i]);
            x=f[x][i];
            y=f[y][i];          
        }
    }
    minn=min(minn,weight[x][0]);
    minn=min(minn,weight[y][0]);
    return minn;
}
int main(){
    scanf("%d%d",&n,&m);
    memset(he,0,sizeof(he));
    for(int i=1;i<=n;i++){
        fa[i]=i;
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].w);
    }
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m;i++){
        hb(e[i].x,e[i].y,e[i].w);
    }
    dep[0]=0;
    for(int i=1;i<=n;i++){
        if(!vis[i]){            
            dfs(i,0,0);
        }   
    }
    csh();
    int q1,q2;
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        scanf("%d%d",&q1,&q2);
        if(find(q1)!=find(q2)){
            printf("-1\n");
            continue;
        }
        printf("%d\n",lca(q1,q2));
    }
    return 0;
}