P1967 货车运输

P1967 货车运输

思路：

　　　　将边权从大到小排序，然后建立最大生成树，在新图上求两个点的lca即可

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cctype>
using namespace std;

#define inf 0x7f7f7f7f
#define LL long long
#define res register int
inline int read()
{
    int x(0),f(1); char ch;
    while(!isdigit(ch=getchar())) if(ch=='-') f=-1;
    while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    return f*x;
}
int n,m;
const int N=500000+5;
int head[N],nxt[N<<1],ver[N<<1],edge[N<<1],tot;
struct E{
    int u,v,w;
}e[N];
int vis[N],fa[N],d[N];
int f[N][22],dis[N][22];

//inline int min(int a,int b){return a<b?a:b;}
inline bool cmp(E a,E b) {return a.w>b.w;}
inline void add(int x,int y,int z) {
    ver[++tot]=y; nxt[tot]=head[x]; head[x]=tot; edge[tot]=z;
    ver[++tot]=x; nxt[tot]=head[y]; head[y]=tot; edge[tot]=z;
}    
//inline void swap(int &x,int &y) {int tmp=x; x=y; y=tmp;}
inline int get(int x) {
    if(x==fa[x]) return x; return fa[x]=get(fa[x]);
}

inline void Kruskal()
{
    sort(e+1,e+m+1,cmp);
    for(res i=1 ; i<=n ; i++) fa[i]=i;
    for(res i=1 ; i<=m ; i++)
    {
        int x=get(e[i].u),y=get(e[i].v),z=e[i].w;
        if(x==y) continue;
        fa[x]=y;
        add(x,y,z);
    }
}

void dfs(int x)
{
    vis[x]=1;
    for(res i(head[x]) ; i ; i=nxt[i])
    {
        int y=ver[i]; 
        if(vis[y]) continue;
        f[y][0]=x; dis[y][0]=edge[i];
//        printf("dis %d",dis[y][0]);
        d[y]=d[x]+1;    dfs(y);
    }
}

inline int lca(int x,int y)
{
    if(get(x)!=get(y)) return -1;
    if(d[x]<d[y]) swap(x,y);
    int ans=inf;
    for(res i=20 ; i>=0 ; i--)
        if(d[f[x][i]]>=d[y])
        {
            ans=min(ans,dis[x][i]);    
            x=f[x][i];
        } 
    if(x==y) return ans;
    for(res i=20 ; i>=0 ; i--)
        if(f[x][i]!=f[y][i])
        {
            ans=min(ans,dis[x][i]);
            ans=min(ans,dis[y][i]);
            x=f[x][i],y=f[y][i];
        }
    ans=min(ans,dis[x][0]);
    ans=min(ans,dis[y][0]);
    return ans;
}


int main()
{
//    freopen("in.txt","r",stdin);
    n=read(); m=read();
    for(res i=1 ; i<=m ; i++) 
        e[i].u=read(), e[i].v=read(),e[i].w=read();
    Kruskal();
    for(res i=1 ; i<=n ; i++)
    {
        if(!vis[i]) 
        {
            d[i]=1; dfs(i);
            f[i][0]=i; dis[i][0]=inf;    
        }
    }
    //注意顺序
    for(int i=1; i<=20; i++)
        for(int j=1; j<=n; j++){
            f[j][i]=f[f[j][i-1]][i-1]; 
            dis[j][i]=min(dis[j][i-1], dis[f[j][i-1]][i-1]);
        }
    int q=read();
    for(res i=1 ; i<=q ; i++)
    {
        int x=read(),y=read();
        printf("%d\n",lca(x,y));
    }
    return 0;
}