洛谷 P1967 货车运输

题目描述
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));
    }
}