洛谷 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：

输出样例#1：

说明

对于 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……

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