[洛谷P1967]货车运输:最小生成树+树链剖分

分析:

很容易想到最大生成森林上的路径肯定满足条件。
于是Kruskal求出最大生成森林后ST表维护树链剖分即可。

代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
inline LL read(){
    LL x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
const int MAXN=100005;
const int MAXM=500005;
int n,m,q,sfa[MAXN];
int ecnt,head[MAXN];
int fa[MAXN],pc[MAXN],siz[MAXN],dep[MAXN],top[MAXN],id[MAXN],w[MAXN],tot;
int st[MAXN][20];
struct Edge{
    int to,nxt,w;
}e[MAXN<<1];
inline void add_edge(int bg,int ed,int val){
    ecnt++;
    e[ecnt].to=ed;
    e[ecnt].nxt=head[bg];
    e[ecnt].w=val;
    head[bg]=ecnt;
}
struct PosEdge{
    int u,v,w;
    friend bool operator < (PosEdge x,PosEdge y){
        return x.w>y.w;
    }
}ee[MAXM];
int getf(int x){
    if(sfa[x]==x) return x;
    return sfa[x]=getf(sfa[x]);
}
inline bool merge(int x,int y){
    int xx=getf(x),yy=getf(y);
    if(xx!=yy){
        sfa[yy]=xx;
        return 1;
    }
    return 0;
}
void dfs1(int x,int pre,int depth){
    fa[x]=pre;
    dep[x]=depth;
    siz[x]=1;
    for(int i=head[x];i;i=e[i].nxt){
        if(e[i].to==pre) continue;
        w[e[i].to]=e[i].w;
        dfs1(e[i].to,x,depth+1);
        siz[x]+=siz[e[i].to];
        if(siz[e[i].to]>siz[pc[x]]){
            pc[x]=e[i].to;
        }
    }
}
void dfs2(int x,int topf){
    top[x]=topf;
    id[x]=++tot;
    st[tot][0]=w[x];
    if(!pc[x]) return;
    dfs2(pc[x],topf);
    for(int i=head[x];i;i=e[i].nxt){
        if(e[i].to==fa[x]||e[i].to==pc[x]) continue;
        dfs2(e[i].to,e[i].to);
    }
}
inline void buildST(){
    int lim=log2(n);
    for(int i=1;i<=lim;i++){
        int len=(1<<i);
        for(int j=1;j+len-1<=n;j++)
            st[j][i]=min(st[j][i-1],st[j+(len>>1)][i-1]);
    }
}
inline int query(int x,int y){
    int lim=log2(y-x+1);
    return min(st[x][lim],st[y-(1<<lim)+1][lim]);
}
inline int pathquery(int x,int y){
    int ans=1e9;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        ans=min(ans,query(id[top[x]],id[x]));
        x=fa[top[x]];
    }
    if(x==y) return ans;
    if(dep[x]>dep[y]) swap(x,y);
    ans=min(ans,query(id[x]+1,id[y]));
    return ans;
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;i++) sfa[i]=i;
    for(int i=1;i<=m;i++){
        ee[i].u=read(),ee[i].v=read(),ee[i].w=read();
    }
    sort(ee+1,ee+m+1);
    int cnt=0;
    for(int i=1;i<=m;i++){
        if(merge(ee[i].u,ee[i].v)){
            cnt++;
            add_edge(ee[i].u,ee[i].v,ee[i].w);
            add_edge(ee[i].v,ee[i].u,ee[i].w);
            if(cnt==n-1) break;
        }
    }
    for(int i=1;i<=n;i++){
        if(id[i]) continue;
        dfs1(i,0,1);
        dfs2(i,i);
    }
    buildST();
    q=read();
    while(q--){
        int u=read(),v=read();
        int uu=getf(u),vv=getf(v);
        if(uu!=vv) printf("-1\n");
        else printf("%d\n",pathquery(u,v));
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ErkkiErkko/p/9609878.html
今日推荐