洛谷 P1967 货车运输 Label: 倍增LCA && 最小瓶颈路

时间:2023-03-08 21:01:29
洛谷 P1967 货车运输 Label:  倍增LCA && 最小瓶颈路

题目描述

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。

代码

 #include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define LEN 100005
#define LOG_LEN 25
#define ll long long
#define INF 0x3f3f3f3f
using namespace std; int n,m;
int fa[LEN],parent[LOG_LEN][LEN],dep[LEN];
ll dis[LOG_LEN][LEN]; struct PA{ll num,d;};//pair
PA make_PA(ll num,ll d){PA tmp;tmp.num=num;tmp.d=d;return tmp;}
struct edge{int from,to;ll val;}e[LEN];
bool cmp(edge a,edge b){return a.val>b.val;}//最大值优先 //BingChaJi ,没有rank版本
void init_BingChaJi(){for(int i=;i<LEN;i++) fa[i]=i;}
int Find(int a){if(fa[a]==a) return a;else return fa[a]=Find(fa[a]);}
void Unite(int a,int b){int pa=Find(a),pb=Find(b);if(pa!=pb) fa[pa]=pb;} vector<PA> Map[LEN]; void Kruskal(){//最大生成树
sort(e,e+m,cmp);
init_BingChaJi();
int cnt=;
for(int i=;i<m;i++){
int a=e[i].from,b=e[i].to;ll v=e[i].val;
if(Find(a)==Find(b))continue;
Unite(a,b);
Map[a].push_back(make_PA(b,v));
Map[b].push_back(make_PA(a,v));
cnt++;
if(cnt==n-) return;
}
} void dfs(int v,int f,int d){
parent[][v]=f;
dis[][v]=INF;
dep[v]=d;
for(int i=;i<Map[v].size();i++){
PA now=Map[v][i];
if(now.num!=f) dfs(now.num,v,d+);
else dis[][v]=now.d;
}
} void init_lca(){
memset(dis,INF,sizeof(dis));
memset(dep,-,sizeof(dep));
for(int i=;i<=n;i++) if(dep[i]<) dfs(i,-,);
for(int k=;k+<LOG_LEN;k++){
for(int i=;i<=n;i++){
if(parent[k][i]<) parent[k+][i]=- , dis[k+][i] = INF;
else{
parent[k+][i]=parent[k][ parent[k][i] ];
dis[k+][i]=min(dis[k][i],dis[k][ parent[k][i] ]);
}
}
}
} ll lca(int u,int v){
if(dep[v]>dep[u]) swap(u,v);
ll res=INF;
for(int i=;i<LOG_LEN;i++){
if( ( (dep[u]-dep[v]) >> i) & ){
res=min(res,dis[i][u]);
u=parent[i][u];
}
}
if(u==v) return res; for(int i=LOG_LEN-;i>=;i--){//这里没有-1会WA
if(parent[i][u]!=parent[i][v]){
res=min(res,dis[i][u]);
res=min(res,dis[i][v]);
u=parent[i][u];v=parent[i][v];
}
}
res=min(res,dis[][u]);
res=min(res,dis[][v]);
return res;
} int main(){
// freopen("01.in","r",stdin);
// freopen("01.out","w",stdout); scanf("%d%d",&n,&m);
for(int i=;i<m;i++) scanf("%d%d%lld",&e[i].from,&e[i].to,&e[i].val); Kruskal();
init_lca(); int kase=;
scanf("%d",&kase);
while(kase--){
int u,v;
scanf("%d%d",&u,&v);
if(Find(u)==Find(v)) printf("%lld\n",lca(v,u));//查找是否同一组
else puts("-1");
}
return ;
}

貌似要用long long

注意Line 80

还有就是我zz了,Line 24

void Unite(int a,int b){int pa=Find(a),pb=Find(b);if(pa!=pb) fa[pa]=pb;}

写成pa==pb,Orz……

抄个代码都花了这么久。。。