BZOJ3732 Network

最小生成树 LCA

题目传送门

刚开始想着二分来着

显然答案在最小生成树上。

然后建完树在树上跳LCA就好了。

代码:

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 15005
#define F inline
using namespace std;
struct edg{ int x,y,d; }e[N<<1];
struct edge{ int nxt,to,d; }ed[N<<1];
int n,m,q,k,h[N],f[N],dep[N],fa[N][16],mx[N][16];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    return l==r?EOF:*l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
F void writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
F void _write(int x){ writec(x),puts(""); }
F bool cmp(edg a,edg b){ return a.d<b.d; }
int findfa(int x){ return x==f[x]?x:f[x]=findfa(f[x]); }
#define addedge(x,y,z) ed[++k]=(edge){h[x],y,z},h[x]=k
F void kruskal(){
    for (int i=1;i<=n;i++) f[i]=i;
    sort(e+1,e+m+1,cmp);
    for (int i=1,x,y;i<=m;i++)
        if (findfa(x=e[i].x)!=findfa(y=e[i].y))
            f[f[x]]=f[y],addedge(x,y,e[i].d),addedge(y,x,e[i].d);
}
void dfs(int x){
    dep[x]=dep[fa[x][0]]+1;
    for (int i=h[x],v;i;i=ed[i].nxt)
        if ((v=ed[i].to)!=fa[x][0])
            fa[v][0]=x,mx[v][0]=ed[i].d,dfs(v);
}
F void Make(){
    for (int j=1;j<=15;j++)
        for (int i=1;i<=n;i++){
            fa[i][j]=fa[fa[i][j-1]][j-1];
            mx[i][j]=max(mx[i][j-1],mx[fa[i][j-1]][j-1]);
        }
}
F int LCA(int x,int y){
    int ans=0;
    if (dep[x]<dep[y]) swap(x,y);
    for (int j=15;~j;j--)
        if (dep[fa[x][j]]>=dep[y])
            ans=max(ans,mx[x][j]),x=fa[x][j];
    if (x==y) return ans;
    for (int j=15;~j;j--)
        if (fa[x][j]!=fa[y][j]){
            ans=max(ans,mx[x][j]),x=fa[x][j];
            ans=max(ans,mx[y][j]),y=fa[y][j];
        }
    return max(ans,max(mx[x][0],mx[y][0]));
}
int main(){
    n=_read(),m=_read(),q=_read();
    for (int i=1,x,y,z;i<=m;i++)
        x=_read(),y=_read(),z=_read(),e[i]=(edg){x,y,z};
    for (kruskal(),dfs(1),Make();q;q--){
        int x=_read(),y=_read();
        printf("%d\n",LCA(x,y));
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/a1799342217/article/details/80949338