bzoj 2238 Mst

显然先求出最小生成树

如果删的是非树边就不用管，还是能取最小生成树

如果删的是树边就有非树边可以替代它

反向考虑，每条非树边可以替代最小生成树上一条路径的边

所以树剖加线段树，对每条非树边在树上更新对应的那一段的答案就行了

代码异常丑陋～～～

#include<bits/stdc++.h>
#define il inline
#define vd void
typedef long long ll;
il int gi(){
    int x=0,f=1;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return x*f;
}
struct edge{int x,y,i,w;}s[100010];
il bool cmp1(const edge&a,const edge&b){return a.w<b.w;}
il bool cmp2(const edge&a,const edge&b){return a.i<b.i;}
int Fa[50010];
il int hd(int x){return Fa[x]==x?x:Fa[x]=hd(Fa[x]);}
il vd Union(int a,int b){Fa[hd(a)]=hd(b);}
int dfn[50010],son[50010],siz[50010],fa[50010],top[50010],dep[50010];
int fir[50010],nxt[100010],dis[100010],id,num[100010];
il vd link(int a,int b,int c){
    nxt[++id]=fir[a],fir[a]=id,dis[id]=b,num[id]=c;
    nxt[++id]=fir[b],fir[b]=id,dis[id]=a,num[id]=c;
}
int NUM[100010];
il vd dfs(int x){
    siz[x]=1;
    for(int i=fir[x];i;i=nxt[i]){
        if(fa[x]==dis[i])continue;
        fa[dis[i]]=x;dep[dis[i]]=dep[x]+1;
        NUM[num[i]]=dis[i];
        dfs(dis[i]);
        siz[x]+=siz[dis[i]];
        if(siz[son[x]]<siz[dis[i]])son[x]=dis[i];
    }
}
il vd dfs2(int x,int tp){
    top[x]=tp;dfn[x]=++dfn[0];
    if(son[x])dfs2(son[x],tp);
    for(int i=fir[x];i;i=nxt[i])
        if(dis[i]!=fa[x]&&dis[i]!=son[x])
            dfs2(dis[i],dis[i]);
}
int lz[200010],ans[100010];
#define mid ((l+r)>>1)
il vd update(int x,int l,int r,const int&L,const int&R,const int&d){
    if(L<=l&&r<=R){lz[x]=std::min(lz[x],d);return;}
    if(L<=mid)update(x<<1,l,mid,L,R,d);
    if(mid<R)update(x<<1|1,mid+1,r,L,R,d);
}
il vd down(int x,int l,int r){
    if(l==r){ans[l]=lz[x];return;}
    lz[x<<1]=std::min(lz[x<<1],lz[x]);
    lz[x<<1|1]=std::min(lz[x<<1|1],lz[x]);
    down(x<<1,l,mid),down(x<<1|1,mid+1,r);
}
int main(){
#ifndef ONLINE_JUDGE
    freopen("2238.in","r",stdin);
    freopen("2238.out","w",stdout);
#endif
    int n=gi(),m=gi();
    for(int i=1;i<=m;++i)s[i].x=gi(),s[i].y=gi(),s[i].w=gi(),s[i].i=i;
    std::sort(s+1,s+m+1,cmp1);
    int q=gi();
    for(int i=1;i<=n;++i)Fa[i]=i;
    int t=1;ll ANS=0;
    for(int i=1;i<n;++i){
        while(t<=m&&hd(s[t].x)==hd(s[t].y))++t;
        if(t>m){
            while(q--)puts("Not connected");
            return 0;
        }
        Union(s[t].x,s[t].y);
        link(s[t].x,s[t].y,s[t].i);
        ANS+=s[t].w;
    }
    std::sort(s+1,s+m+1,cmp2);
    dfs(1),dfs2(1,1);
    for(int i=1;i<=n*4;++i)lz[i]=1e9;
    for(int i=1;i<=m;++i)
        if(!NUM[i]){
            int x=s[i].x,y=s[i].y;
            while(top[x]!=top[y]){
                if(dep[top[x]]>dep[top[y]])update(1,1,n,dfn[top[x]],dfn[x],s[i].w),x=fa[top[x]];
                else update(1,1,n,dfn[top[y]],dfn[y],s[i].w),y=fa[top[y]];
            }
            if(dep[x]>dep[y])std::swap(x,y);
            update(1,1,n,dfn[x]+1,dfn[y],s[i].w);
        }
    down(1,1,n);
    while(q--){
        int x=gi();
        if(NUM[x]){
            if(ans[dfn[NUM[x]]]==1e9)puts("Not connected");
            else printf("%lld\n",ANS-s[x].w+ans[dfn[NUM[x]]]);
        }else printf("%lld\n",ANS);
    }
    return 0;
}

转载于:https://www.cnblogs.com/xzz_233/p/9722008.html