题目描述
A 国有 n 座城市,编号从 1 到 n,城市之间有 m 条双向道路。每一条道路对车辆都有重量限制,简称限重。现在有 q 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
【题目分析】
最大生成树+LCA
【代码】
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
struct road{int l,r,w;}a[50001];
int n,m,q,l,r,fa[10001];
int h[10001],to[20001],ne[20001],v[20001],en=0;
int f[10001][16],g[10001][16],vis[10001];
int fat[10001],dep[10001];
int read()
{
int ret=0,f=1; char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){ret*=10;ret+=ch-'0';ch=getchar();}
return ret*f;
}
int gf(int k)
{
if (fa[k]==k) return k;
else return fa[k]=gf(fa[k]);
}
bool cmp(road a,road b)
{return a.w>b.w;}
void add(int a,int b,int c)
{to[en]=b;ne[en]=h[a];v[en]=c;h[a]=en++;}
void dfs(int k)
{
vis[k]=1;
for (int i=h[k];i>=0;i=ne[i])
{
if (to[i]!=f[k][0])
{
f[to[i]][0]=k;
g[to[i]][0]=v[i];
dep[to[i]]=dep[k]+1;
dfs(to[i]);
}
}
}
void init()
{
for (int i=1;i<=15;++i)
for (int j=1;j<=n;++j)
f[j][i]=f[f[j][i-1]][i-1],
g[j][i]=min(g[j][i-1],g[f[j][i-1]][i-1]);
}
int lca(int a,int b)
{
int ans=0x3f3f3f3f;
if (dep[a]<dep[b]) swap(a,b);
int def=dep[a]-dep[b];
for (int i=15;i>=0;--i)
if (def&(1<<i))
{
ans=min(ans,g[a][i]);
a=f[a][i];
}
if (a==b) return ans;
for (int i=15;i>=0;--i)
if (f[a][i]!=f[b][i])
{
ans=min(ans,g[a][i]);
a=f[a][i];
ans=min(ans,g[b][i]);
b=f[b][i];
}
ans=min(ans,min(g[a][0],g[b][0]));
return ans;
}
int main()
{
memset(h,-1,sizeof h);
n=read();m=read();
for (int i=1;i<=n;++i) fa[i]=i;
for (int i=1;i<=m;++i) a[i].l=read(),a[i].r=read(),a[i].w=read();
sort(a+1,a+m+1,cmp);
for (int i=1;i<=m;++i)
{
int fl=gf(a[i].l),fr=gf(a[i].r);
if (fl!=fr)
{
fa[fl]=fr;
add(a[i].l,a[i].r,a[i].w);
add(a[i].r,a[i].l,a[i].w);
}
}
for (int i=1;i<=n;++i) if (!vis[i]) dfs(i);
init();
q=read();
while (q--)
{
l=read();r=read();
int fl=gf(l),fr=gf(r);
if (fl!=fr) printf("-1\n");
else printf("%d\n",lca(l,r));
}
}