NOIP 2013 货车运输

题目描述

$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<1000,0<m<10000,0<q<1000$

对于 $60\%$ 的数据，$0<n<1000,0<m<50000,0<q<1000$

对于 $100\%$ 的数据，$0<n<10000,0<m<50000,0<q<30000,0\leq z \leq 100000$

---

solution

• 先用Kruskal跑一遍最大生成树，同时建出一棵树，然后在这个树上跑倍增lca统计答案

code

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
#define MAXN 10010
#define MAXM 50010

template<typename T>
void input(T &x) {
    x=0; T a=1;
    register char c=getchar();
    for(;c<'0'||c>'9';c=getchar())
        if(c=='-') a=-1;
    for(;c>='0'&&c<='9';c=getchar())
        x=x*10+c-'0';
    x*=a;
    return;
}

struct Edge {

    int u,v,w,next;

    Edge(int u=0,int v=0,int w=0,int next=0):
        u(u),v(v),w(w),next(next) {}

    bool operator < (const Edge &q)const {
        return w>q.w;
    }

}edge[MAXN<<1],e[MAXM];
int head[MAXN],cnt;

void addedge(int u,int v,int w) {
    edge[cnt]=Edge(u,v,w,head[u]);
    head[u]=cnt++;
    return;
}

int father[MAXN],n,m;

int Find(int x) {
    return father[x]==x?x:father[x]=Find(father[x]);
}

void Kruskal_Build_tree() {
    for(int i=1;i<=n;i++) father[i]=i,head[i]=-1;
    sort(e+1,e+m+1);
    for(int i=1;i<=m;i++) {
        int fu=Find(e[i].u),fv=Find(e[i].v);
        if(fu!=fv) {
            addedge(e[i].u,e[i].v,e[i].w);
            addedge(e[i].v,e[i].u,e[i].w);
            father[fu]=fv;
        }
    }
    return;
}

int depth[MAXN];
int fa[MAXN][31],dis[MAXN][31];

void dfs(int u,int p) {
    for(int i=head[u];~i;i=edge[i].next)
        if(i!=p) {
            int v=edge[i].v;
            depth[v]=depth[u]+1;
            fa[v][0]=u;
            dis[v][0]=edge[i].w;
            dfs(v,i^1);
        }
    return;
}

int lca(int x,int y) {
    int ans=~0u>>1;
    if(depth[x]>depth[y]) swap(x,y);
    for(int i=15;i>=0;i--)
        if(depth[fa[y][i]]>=depth[x]) {
            ans=min(ans,dis[y][i]);
            y=fa[y][i];
        }
    if(x==y) return ans;
    for(int i=15;i>=0;i--)
        if(fa[x][i]!=fa[y][i]) {
            ans=min(ans,min(dis[x][i],dis[y][i]));
            x=fa[x][i];
            y=fa[y][i];
        }
    if(!fa[x][0]) return -1;
    return min(ans,min(dis[x][0],dis[y][0]));
}

int main() {
    input(n),input(m);
    for(int i=1;i<=m;i++)
        input(e[i].u),input(e[i].v),input(e[i].w);
    Kruskal_Build_tree();
    dfs(depth[1]=1,-1);
    for(int j=1;j<=15;j++)
        for(int i=1;i<=n;i++) {
            fa[i][j]=fa[fa[i][j-1]][j-1];
            dis[i][j]=min(dis[i][j-1],dis[fa[i][j-1]][j-1]);
        }
    int Q;
    input(Q);
    while(Q--) {
        int x,y;
        input(x),input(y);
        printf("%d\n",lca(x,y));
    }
    return 0;
}