【模板】P1967 货车运输(最小生成树+树上lca)


题目链接:https://www.luogu.com.cn/problem/P1967

题目描述:

A 国有 nn 座城市,编号从 11 到 nn,城市之间有 mm 条双向道路。每一条道路对车辆都有重量限制,简称限重。
现在有 qq 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
输入格式
第一行有两个用一个空格隔开的整数 n,mn,m,表示 AA 国有 nn 座城市和 mm 条道路。
接下来 mm 行每行三个整数 x, y, zx,y,z,每两个整数之间用一个空格隔开,表示从 xx 号城市到 yy 号城市有一条限重为 zz 的道路。
注意: 两座城市之间可能有多条道路 。
接下来一行有一个整数 qq,表示有 qq 辆货车需要运货。
接下来 qq 行,每行两个整数 x,yx,y,之间用一个空格隔开,表示一辆货车需要从 xx 城市运输货物到 yy 城市,保证 x !=y
输出格式
共有 qq 行,每行一个整数,表示对于每一辆货车,它的最大载重是多少。
如果货车不能到达目的地,输出 -1−1。

#include<bits/stdc++.h>
#define mp_ make_pair
#define Pr pair<int,int>
using namespace std;
typedef long long ll;

const int NN=20010;
const int inf=0x3f3f3f3f;
ll n,m,k,x,y,z,q,W,T,N,cnt,tmp,num,siz,ans,dst;
string s;char ch;

ll fa[NN],dep[NN],w[NN][25],lca[NN][25];
vector<Pr> ve[NN];
pair<int,Pr> edge[NN*3];

//并查集
int father(int x){
    if(fa[x]==x) return x;
    return fa[x]=father(fa[x]);
}

//求最大生成树,建图
void kruscal(){
    sort(edge+1,edge+num+1);
    for(int i=num;i>=1;i--){
        Pr p=edge[i].second;
        if(father(p.first)==father(p.second)) continue;
        fa[father(p.first)]=father(p.second);
        ve[p.first].push_back(mp_(p.second,edge[i].first));
        ve[p.second].push_back(mp_(p.first,edge[i].first));
    }
}

//以某节点为根结点刷新层次
void build(int u,int f){
    dep[u]=dep[f]+1;lca[u][0]=f;
    for(auto &v:ve[u]) if(v.first!=f) w[v.first][0]=v.second,build(v.first,u);
}

ll min(ll a,ll b,ll c){return min(a,min(b,c));}

ll LCA(int x,int y,ll ans=inf){
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=20;i>=0;i--){
        if(dep[lca[x][i]]>=dep[y]){
            ans=min(ans,w[x][i]);
            x=lca[x][i];
        }
    }
    if(x==y) return ans;
    for(int i=20;i>=0;i--){
        if(lca[x][i]!=lca[y][i]){
            ans=min(ans,w[x][i],w[y][i]);
            x=lca[x][i],y=lca[y][i];
        }
    }
    return min(ans,w[x][0],w[y][0]);
}

int main(){
    ios::sync_with_stdio(false);//cin.tie(0);cout.tie(0);
    //freopen("testdata.txt","r",stdin);
    
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>x>>y>>z;fa[x]=x,fa[y]=y;
        edge[++num]=mp_(z,mp_(x,y));
    }
    kruscal();
    
    //注意这里可能有不连通的几棵树
    for(int i=1;i<=n;i++){if(dep[i]==0) build(i,0),w[i][0]=inf;}

	//预处理lca
    for(int j=1;j<=20;j++){
        for(int i=1;i<=n;i++){
            lca[i][j]=lca[lca[i][j-1]][j-1];
            if(dep[i]>1) w[i][j]=min(w[i][j-1],w[lca[i][j-1]][j-1]);
        }
    }
    cin>>q;
    while(q--){
        cin>>x>>y;
        
        //不在同一棵树中
        if(father(x)!=father(y)) cout<<-1<<endl;
        else cout<<LCA(x,y)<<endl;
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值