首页 > 代码库 > 洛谷 P1967 货车运输 Label: 倍增LCA && 最小瓶颈路
洛谷 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。
输入输出样例
4 31 2 42 3 33 1 131 31 41 3
3-13
说明
对于 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。
代码
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<vector> 6 #define LEN 100005 7 #define LOG_LEN 25 8 #define ll long long 9 #define INF 0x3f3f3f3f 10 using namespace std; 11 12 int n,m; 13 int fa[LEN],parent[LOG_LEN][LEN],dep[LEN]; 14 ll dis[LOG_LEN][LEN]; 15 16 struct PA{ll num,d;};//pair 17 PA make_PA(ll num,ll d){PA tmp;tmp.num=num;tmp.d=d;return tmp;} 18 struct edge{int from,to;ll val;}e[LEN]; 19 bool cmp(edge a,edge b){return a.val>b.val;}//最大值优先 20 21 //BingChaJi ,没有rank版本 22 void init_BingChaJi(){for(int i=0;i<LEN;i++) fa[i]=i;} 23 int Find(int a){if(fa[a]==a) return a;else return fa[a]=Find(fa[a]);} 24 void Unite(int a,int b){int pa=Find(a),pb=Find(b);if(pa!=pb) fa[pa]=pb;} 25 26 vector<PA> Map[LEN]; 27 28 void Kruskal(){//最大生成树 29 sort(e,e+m,cmp); 30 init_BingChaJi(); 31 int cnt=0; 32 for(int i=0;i<m;i++){ 33 int a=e[i].from,b=e[i].to;ll v=e[i].val; 34 if(Find(a)==Find(b))continue; 35 Unite(a,b); 36 Map[a].push_back(make_PA(b,v)); 37 Map[b].push_back(make_PA(a,v)); 38 cnt++; 39 if(cnt==n-1) return; 40 } 41 } 42 43 void dfs(int v,int f,int d){ 44 parent[0][v]=f; 45 dis[0][v]=INF; 46 dep[v]=d; 47 for(int i=0;i<Map[v].size();i++){ 48 PA now=Map[v][i]; 49 if(now.num!=f) dfs(now.num,v,d+1); 50 else dis[0][v]=now.d; 51 } 52 } 53 54 void init_lca(){ 55 memset(dis,INF,sizeof(dis)); 56 memset(dep,-1,sizeof(dep)); 57 for(int i=1;i<=n;i++) if(dep[i]<0) dfs(i,-1,0); 58 for(int k=0;k+1<LOG_LEN;k++){ 59 for(int i=1;i<=n;i++){ 60 if(parent[k][i]<0) parent[k+1][i]=-1 , dis[k+1][i] = INF; 61 else{ 62 parent[k+1][i]=parent[k][ parent[k][i] ]; 63 dis[k+1][i]=min(dis[k][i],dis[k][ parent[k][i] ]); 64 } 65 } 66 } 67 } 68 69 ll lca(int u,int v){ 70 if(dep[v]>dep[u]) swap(u,v); 71 ll res=INF; 72 for(int i=0;i<LOG_LEN;i++){ 73 if( ( (dep[u]-dep[v]) >> i) & 1 ){ 74 res=min(res,dis[i][u]); 75 u=parent[i][u]; 76 } 77 } 78 if(u==v) return res; 79 80 for(int i=LOG_LEN-1;i>=0;i--){//这里没有-1会WA 81 if(parent[i][u]!=parent[i][v]){ 82 res=min(res,dis[i][u]); 83 res=min(res,dis[i][v]); 84 u=parent[i][u];v=parent[i][v]; 85 } 86 } 87 res=min(res,dis[0][u]); 88 res=min(res,dis[0][v]); 89 return res; 90 } 91 92 int main(){ 93 // freopen("01.in","r",stdin); 94 // freopen("01.out","w",stdout); 95 96 scanf("%d%d",&n,&m); 97 for(int i=0;i<m;i++) scanf("%d%d%lld",&e[i].from,&e[i].to,&e[i].val); 98 99 Kruskal();100 init_lca();101 102 int kase=0;103 scanf("%d",&kase);104 while(kase--){105 int u,v;106 scanf("%d%d",&u,&v);107 if(Find(u)==Find(v)) printf("%lld\n",lca(v,u));//查找是否同一组 108 else puts("-1");109 }110 return 0;111 }貌似要用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……
抄个代码都花了这么久。。。
洛谷 P1967 货车运输 Label: 倍增LCA && 最小瓶颈路