洛谷 P1967 货车运输

题目描述

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。

分析

先做一遍最大生成树，然后对于任意询问倍增求一下lca，再随便查询一下答案就出来了。

代码

#include<cstdio>  
#include<cstring>  
#include<cmath>  
#include<algorithm>  
#include<vector>  
#include<queue>  
#define rep(i,l,r) for(int i=l;i<=r;i++)  
#define repu(i,r,l) for (int i=r;i>=l;i--)  
#define N 10100  
#define M 50500  
using namespace std;  
struct data  
{  
    int u,v,w;  
}edge[M];  
struct way  
{  
    int v,w,next;  
};  
int n,m,q,p[N],dep[N],parent[N][17],afford[N][17],x,y,z,head[N],cnt;  
bool vis[N];  
way map[2*M];  
bool cmp(const data &a,const data &b)  
{  
    return a.w>b.w;  
}  
void ufsets()  
{  
    rep(i,1,n)  
    p[i]=-1;  
}  
void addedge(int x,int y,int value)  
{  
    way e;  
    cnt++;  
    e.v=y;  
    e.w=value;  
    e.next=head[x];  
    head[x]=cnt;  
    map[cnt]=e;  
    cnt++;  
    e.v=x;  
    e.next=head[y];  
    head[y]=cnt;  
    map[cnt]=e;  
}  
int find(int x)  
{  
    return p[x]<0?x:p[x]=find(p[x]);  
}  
void uni(int x,int y)  
{  
    if (p[x]<=p[y])  
    {  
        p[x]+=p[y];  
        p[y]=x;  
    }  
    else  
    {  
        p[y]+=p[x];  
        p[x]=y;  
    }  
}  
void dfs(int x,int pa,int d)  
{  
    parent[x][0]=pa;  
    dep[x]=d;  
    vis[x]=true;  
    for (int i=head[x];i;i=map[i].next)  
    {  
        int t=map[i].v;  
        if ((!vis[t])&&(t!=x))dfs(t,x,d+1);  
    }  
}  
int neiaff(int x,int y)  
{  
    for (int i=head[x];i;i=map[i].next)  
    if (map[i].v==y) return map[i].w;  
}  
void makemap()  
{  
    memset(vis,0,sizeof(vis));  
    memset(parent,0,sizeof(parent));  
    int tree=0;  
    rep(i,1,n)  
    {  
        if (!vis[i])  
        {  
            tree--;  
            dfs(i,tree,0);  
        }  
    }  
    rep(j,1,16)  
    {  
        rep(i,1,n)  
        {  
            if (parent[i][j-1]<0) parent[i][j]=parent[i][j-1];  
            parent[i][j]=parent[parent[i][j-1]][j-1];  
        }  
    }  
    rep(i,1,n)  
    {  
        afford[i][0]=neiaff(i,parent[i][0]);  
    }  
    rep(j,1,16)  
    {  
        rep(i,1,n)  
        {  
            if (parent[i][j-1]<0)continue;  
            afford[i][j]=min(afford[i][j-1],afford[parent[i][j-1]][j-1]);  
        }  
    }  
}  
int getaff(int x,int y)  
{  
    int t=dep[x]-dep[y];  
    int ans=1234567890;  
    rep(i,0,16)  
    if ((t>>i)&1)  
    {     
        ans=min(ans,afford[x][i]);  
        x=parent[x][i];   
    }     
    return ans;  
}  
int lca(int x,int y)  
{  
    if (dep[x]<dep[y]) swap(x,y);  
    int t=dep[x]-dep[y];  
    rep(i,0,16)  
    if (t&(1<<i)) x=parent[x][i];  
    repu(i,15,0)  
    if (parent[x][i]!=parent[y][i])  
    {  
        x=parent[x][i];  
        y=parent[y][i];  
    }  
    if (x==y) return x;  
    else return parent[x][0];  
}  
int main()  
{  
    scanf("%d%d",&n,&m);  
    rep(i,1,m)  
    scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);  
    sort(edge+1,edge+m+1,cmp);  
    cnt=0;  
    ufsets();  
    memset(head,0,sizeof(head));  
    rep(i,1,m)  
    {  
        x=find(edge[i].u);  
        y=find(edge[i].v);  
        if (x!=y)  
        {  
            uni(x,y);  
            addedge(x,y,edge[i].w);  
        }  
    }  
    makemap();  
    scanf("%d",&q);  
    rep(i,1,q)  
    {  
        scanf("%d%d",&x,&y);  
        if (find(x)!=find(y))  
        {  
            printf("-1\n");  
            continue;  
        }  
        else  
        {  
            int anc=lca(x,y);  
            printf("%d\n",min(getaff(x,anc),getaff(y,anc)));  
        }  
    }  
    return 0;  
}