zoukankan      html  css  js  c++  java
  • 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
    ",ANS-s[x].w+ans[dfn[NUM[x]]]);
    		}else printf("%lld
    ",ANS);
    	}
    	return 0;
    }
    
  • 相关阅读:
    fastJson
    关于mybatis “org.apache.ibatis.binding.BindingException: Invalid bound statement (not found)” 错误的问题。
    JVM——自定义类加载器
    NumberFormat数字格式化
    Java BigDecimal详解
    NumberUtils、ArrayUtils和RandomUtils工具类用法
    itext 生成 PDF(二)
    C语言中的可变参数列表
    归并排序--c语言实现
    机器学习实战学习笔记 一 k-近邻算法
  • 原文地址:https://www.cnblogs.com/xzz_233/p/9722008.html
Copyright © 2011-2022 走看看