货车运输
题目描述
A 国有 n 座城市,编号从 1 到 n,城市之间有 m 条双向道路。每一条道路对车辆都有重量限制,简称限重。
现在有 q 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
输入格式
第一行有两个用一个空格隔开的整数 n,m,表示 A 国有 n 座城市和 m 条道路。
接下来 m 行每行三个整数 x,y,z,每两个整数之间用一个空格隔开,表示从 x 号城市到 y 号城市有一条限重为 z 的道路。
注意: x≠y,两座城市之间可能有多条道路 。
接下来一行有一个整数 q,表示有 q 辆货车需要运货。
接下来 q 行,每行两个整数 x,y,之间用一个空格隔开,表示一辆货车需要从 x 城市运输货物到 y 城市,保证 x≠y
输出格式
共有 q 行,每行一个整数,表示对于每一辆货车,它的最大载重是多少。
如果货车不能到达目的地,输出 −1。
样例数据
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3
3
-1
3
数据范围
对于 30% 的数据,1≤n<1000,1≤m<10,000,1≤q<1000;
对于 60% 的数据,1≤n<1000,1≤m<5×104,1≤q<1000;
对于 100% 的数据,1≤n<104,1≤m<5×104,1≤q<3×104,0≤z≤105。
正解:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
using namespace std;
const int maxn=10010;
const int maxm=50010;
const int INF=1e9;
struct edge
{
int u,v,w;
}e[maxm];
int head[maxn],nnext[maxm*2],to[maxm*2],length[maxm*2];
int depth[maxn],f[maxn],fa[maxn][30],minn[maxn][30];
bool b[maxn];
int tot=0,n,m,q;
void add(int x,int y,int l)
{
tot++;
nnext[tot]=head[x];
head[x]=tot;
to[tot]=y;
length[tot]=l;
}
bool cmp(const edge &a,const edge &b)
{
return a.w>b.w;
}
int find(int x)
{
if(x!=f[x]) f[x]=find(f[x]);
return f[x];
}
void kruskal()
{
sort(e+1,e+m+1,cmp);
for(int i=1;i<=n;i++)
{
f[i]=i;
}
for(int i=1;i<=m;i++)
{
int x1=e[i].u;
int x2=e[i].v;
int f1=find(x1);
int f2=find(x2);
if(f1!=f2)
{
f[f1]=f2;
add(x1,x2,e[i].w);
add(x2,x1,e[i].w);
}
}
return ;
}
void dfs(int x)
{
b[x]=true;
for(int i=head[x];i;i=nnext[i])
{
int y=to[i];
if(b[y]==true) continue;
depth[y]=depth[x]+1;
fa[y][0]=x;
minn[y][0]=length[i];
dfs(y);
}
return ;
}
int lca(int x,int y)
{
if(find(x)!=find(y)) return -1;
int ans=INF;
if(depth[x]>depth[y]) swap(x,y);
for(int i=20;i>=0;i--)
{
if(depth[fa[y][i]]>=depth[x])
{
ans=min(ans,minn[y][i]);
y=fa[y][i];
}
}
if(x==y) return ans;
for(int i=20;i>=0;i--)
{
if(fa[x][i]!=fa[y][i])
{
ans=min(ans,min(minn[x][i],minn[y][i]));
x=fa[x][i];
y=fa[y][i];
}
}
//cout<<ans<<' '<<x<<' '<<y<<' '<<minn[x][0]<<' '<<minn[y][0]<<' '<<endl;
ans=min(ans,min(minn[x][0],minn[y][0]));
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
}
kruskal();
for(int i=1;i<=n;i++)
{
if(!b[i])
{
depth[i]=1;
dfs(i);
}
}
for(int i=1;i<=20;i++)
{
for(int j=1;j<=n;j++)
{
fa[j][i]=fa[fa[j][i-1]][i-1];
minn[j][i]=min(minn[j][i-1],minn[fa[j][i-1]][i-1]);
}
}
scanf("%d",&q);
for(int i=1;i<=q;i++)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",lca(x,y));
}
return 0;
}