题目描述
A 国有 n 座城市,编号从 1 到 n,城市之间有 m 条双向道路。每一条道路对车辆都有重量限制,简称限重。现在有 q 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
【题目分析】
最大生成树+倍增求LCA,也是一道水题。
【代码】
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
struct line{
int l,r,v;
}a[50001];
int fa[10001];
int n,m,q;
int h[10001],to[50001],w[50001],ne[50001],en=0;
int f[16][10001],g[16][10001],dep[10001];
inline void add(int a,int b,int c)
{to[en]=b;w[en]=c;ne[en]=h[a];h[a]=en++;}
inline int cmp(line a,line b)
{return a.v>b.v;}
inline int gf(int k)
{
if (fa[k]==k) return k;
else return fa[k]=gf(fa[k]);
}
inline void dfs(int k,int de,int fat,int fw)
{
f[0][k]=fat;
g[0][k]=fw;
dep[k]=de;
for (int i=h[k];i>=0;i=ne[i])
if (!dep[to[i]]) dfs(to[i],de+1,k,w[i]);
}
inline void init()
{
for (int i=1;i<=15;++i)
for (int j=1;j<=n;++j)
f[i][j]=f[i-1][f[i-1][j]],g[i][j]=min(g[i-1][j],g[i-1][f[i-1][j]]);
}
inline void lca(int a,int b)
{
int ans=0x3f3f3f3f;
if (dep[a]<dep[b]) swap(a,b);
for (int i=15;i>=0;--i)
if (dep[a]>=dep[b]+(1<<i)) ans=min(ans,g[i][a]),a=f[i][a];
if (a==b) {cout<<ans<<endl;return ;}
for (int i=15;i>=0;--i)
if (f[i][a]!=f[i][b])
{
ans=min(ans,min(g[i][a],g[i][b]));
a=f[i][a];b=f[i][b];
}
ans=min(ans,min(g[0][a],g[0][b]));
cout<<ans<<endl;
return ;
}
int main()
{
memset(g,0x3f,sizeof g);
memset(h,-1,sizeof h);
scanf("%d%d",&n,&m);
for (int i=1;i<=n;++i) fa[i]=i;
for (int i=1;i<=m;++i)
scanf("%d%d%d",&a[i].l,&a[i].r,&a[i].v);
sort(a+1,a+m+1,cmp);
// cout<<a[1].l<<" "<<a[1].r<<" "<<a[1].v<<endl;
for (int i=1;i<=m;++i)
{
int fl=gf(a[i].l),fr=gf(a[i].r);
if (fl!=fr) add(fl,fr,a[i].v),add(fr,fl,a[i].v),fa[fl]=fr;
}
scanf("%d",&q);
dfs(1,1,0,0);
init();
for (int i=1;i<=q;++i)
{
int x,y;
scanf("%d%d",&x,&y);
int fx=gf(x),fy=gf(y);
if (fx!=fy) {cout<<"-1"<<endl; continue;}
else lca(x,y);
}
}