2001: [Hnoi2010]City 城市建设

最新推荐文章于 2024-08-28 09:31:48 发布

wamach

最新推荐文章于 2024-08-28 09:31:48 发布

阅读量435

点赞数

CC 4.0 BY-SA版权

分类专栏： BZOJ

本文链接：https://blog.youkuaiyun.com/wamach/article/details/76461388

BZOJ 专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种基于分治思想的最小生成树(MST)算法实现细节，该算法通过递归地将问题分解为更小的部分来寻找最优解。在每个阶段，算法通过修改边的权重来确定哪些边是必要的，哪些可以被删除。最终目的是找到包含所有顶点的最小权重边集合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

当分治到区间[l,r]时：

如果l==r，直接跑mst，结束。否则：

S1存初始边和[1,l)的边，S2存[l,r]的边。

先把S2中的边权全都改成inf，跑mst，S1中用不到的边即为无用边，直接删去。

再把S2中的边权全都改成-inf，跑mst，S1中仍被用到的边即为必需边，把这些边连接的两点用并查集搞在一起，并把这些边权和传到下一层。

对于分治的每一种深度，开一个边、一个点集，一个并查集，一坨询问。因为深度最多为logn、同种深度的区间不会同时运行，所以只需开logn个就够了，空间是不会爆的。

TMD把题意看错了，写了个只有加边的。。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
#define rep(i,j,k) for(i=j;i<=k;++i)
#define per(i,j,k) for(i=j;i>=k;--i)
#define sqr(x) ((x)*(x))
#define G getchar()
#define LL long long
#define pii pair<int,int>
#define mkp make_pair
#define X first
#define Y second
#define N 200005
#define NN 500005
#define inf 100000000
struct EDGE{int a,b,w;}qu[NN];
struct EDGE2{int a,b,w,wz;}ek[NN<<1];
struct node{
	EDGE e[NN<<1];
	int tot,poi[N],cnt,par[N],sz[N];
}Hz[17];
int n,m,Q,totk,park[N],szk[N];
LL ans[NN];
bool flg[NN<<1];
void read(int &x){
	char ch=G;
	while(ch<48||ch>57)ch=G;
	for(x=0;ch>47&&ch<58;ch=G)x=x*10+ch-48; 
}
bool cmp(EDGE2 x,EDGE2 y){return x.w<y.w;}
int getpar(int *par0,int x){
	if(par0[x]!=x)par0[x]=getpar(par0,par0[x]);
	return par0[x];
}
void cmb(int *par0,int *sz0,int x,int y){
	x=getpar(par0,x);y=getpar(par0,y);
	if(sz0[x]<sz0[y])par0[x]=par0[y];
	else par0[y]=par0[x];
}
LL kruscal(int *poi0,int cnt0){
	int i,x,y,j;LL rtn=0;
	rep(i,1,cnt0)szk[park[i]=i]=1;
	rep(i,1,totk)flg[i]=0;
	sort(ek+1,ek+totk+1,cmp);
	for(i=j=1;j<cnt0;++i){
		x=getpar(park,ek[i].a);y=getpar(park,ek[i].b);
		if(x==y)continue;++j;
		flg[ek[i].wz]=1;rtn+=ek[i].w;
		cmb(park,szk,ek[i].a,ek[i].b);
	}
	return rtn;
}
void cdq(int l,int r,LL sum,int dep){
	EDGE *e0=Hz[dep].e;
	int &tot0=Hz[dep].tot,*poi0=Hz[dep].poi,&cnt0=Hz[dep].cnt,*par0=Hz[dep].par,*sz0=Hz[dep].sz;
	int i;
	rep(i,1,cnt0)sz0[par0[i]=i]=1;
	totk=tot0;
	rep(i,1,tot0)ek[i]=(EDGE2){e0[i].a,e0[i].b,e0[i].w,i};
	if(l==r){
		ek[++totk]=(EDGE2){qu[l].a,qu[l].b,qu[l].w,totk};
		ans[l]=kruscal(poi0,cnt0)+sum;
		return;
	}
	rep(i,l,r)ek[++totk]=(EDGE2){qu[i].a,qu[i].b,inf,totk};
	kruscal(poi0,cnt0);
	totk=tot0;
	rep(i,1,tot0){
		ek[i]=(EDGE2){e0[i].a,e0[i].b,e0[i].w,i};
		if(!flg[i])e0[i].a=0;
	}
	rep(i,l,r)ek[++totk]=(EDGE2){qu[i].a,qu[i].b,-inf,totk};
	kruscal(poi0,cnt0);
	EDGE *e1=Hz[dep+1].e;
	int &tot1=Hz[dep+1].tot,*poi1=Hz[dep+1].poi,&cnt1=Hz[dep+1].cnt;
	int x,mid=l+r>>1;
	rep(i,1,tot0){
		if(!e0[i].a)continue;
		if(flg[i]){
			cmb(par0,sz0,e0[i].a,e0[i].b);
			e0[i].a=0;sum+=e0[i].w;
		}
	}
	rep(i,1,cnt0){
		x=poi0[i];
		if(getpar(par0,x)==x)poi1[++cnt1]=x;
	}
	rep(i,1,tot0)if(e0[i].a)
		e1[++tot1]=(EDGE){getpar(par0,e0[i].a),getpar(par0,e0[i].b),e0[i].w};
	rep(i,l,r){
		qu[i].a=getpar(par0,qu[i].a);
		qu[i].b=getpar(par0,qu[i].b);
	}
	cdq(l,mid,sum,dep+1);
	rep(i,l,mid)e1[++tot1]=qu[i];
	cdq(mid+1,r,sum,dep+1);
}
int main(){
	int i,x;
	read(n);read(m);read(Q);
	EDGE *e0=Hz[1].e;
	rep(i,1,m){
		read(e0[i].a);read(e0[i].b);read(e0[i].w);
	}
	Hz[1].tot=m;
	int *poi0=Hz[1].poi;
	rep(i,1,n)poi0[i]=i;
	Hz[1].cnt=n;
	rep(i,1,Q){
		read(x);read(qu[i].w);
		qu[i].a=e0[x].a;qu[i].b=e0[x].b;
	}
	cdq(1,Q,0LL,1);
	rep(i,1,Q)printf("%d\n",ans[i]);
	return 0;
}

简直脑子进水，下面的代码写了我一整天。不过我还真没见过这么难写的题。~~觉得这题比Claris loves painting 还难写得多。~~

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
#define rep(i,j,k) for(i=j;i<=k;++i)
#define per(i,j,k) for(i=j;i>=k;--i)
#define sqr(x) ((x)*(x))
#define G getchar()
#define LL long long
#define pii pair<int,int>
#define mkp make_pair
#define X first
#define Y second
#define N 20005
#define NN 50005
#define inf 100000000
struct EDGE{int a,b,w;};
struct EDGE2{int a,b,w,wz;}ek[NN<<1];
struct QUERY{int cur,w;};
struct node{
	EDGE e[NN<<1];QUERY qu[NN];
	int tot,poi[N],cnt,par[N],sz[N];
}Hz[20];
int n,m,Q,totk,park[N],szk[N],key[NN];
LL ans[NN];
bool flg[NN<<1];
void read(int &x){
	char ch=G;
	while(ch<48||ch>57)ch=G;
	for(x=0;ch>47&&ch<58;ch=G)x=x*10+ch-48; 
}
bool cmp(EDGE2 x,EDGE2 y){return x.w<y.w;}
int getpar(int *par0,int x){
	if(par0[x]!=x)par0[x]=getpar(par0,par0[x]);
	return par0[x];
}
void cmb(int *par0,int *sz0,int x,int y){
	x=getpar(par0,x);y=getpar(par0,y);
	if(sz0[x]<sz0[y])sz0[par0[x]=y]+=sz0[x];
	else sz0[par0[y]=x]+=sz0[y];
}
LL kruscal(int *poi0,int cnt0){
	int i,x,y,j;LL rtn=0;
	rep(i,1,cnt0)szk[park[poi0[i]]=poi0[i]]=1;
	rep(i,1,totk)flg[i]=0;
	sort(ek+1,ek+totk+1,cmp);
	for(i=j=1;j<cnt0;++i){
		x=getpar(park,ek[i].a);y=getpar(park,ek[i].b);
		if(x==y)continue;++j;
		flg[ek[i].wz]=1;rtn+=ek[i].w;
		cmb(park,szk,ek[i].a,ek[i].b);
	}
	return rtn;
}
void cdq(int l,int r,LL sum,int dep){
	EDGE *e0=Hz[dep].e;QUERY *qu0=Hz[dep].qu;
	int &tot0=Hz[dep].tot,*poi0=Hz[dep].poi,&cnt0=Hz[dep].cnt,*par0=Hz[dep].par,*sz0=Hz[dep].sz;
	int i;
	rep(i,1,cnt0)sz0[par0[poi0[i]]=poi0[i]]=1;
	totk=tot0;
	rep(i,1,tot0)ek[i]=(EDGE2){e0[i].a,e0[i].b,e0[i].w,i};
	if(l==r){
		ek[qu0[l].cur].w=qu0[l].w;
		ans[l]=kruscal(poi0,cnt0)+sum;
		return;
	}
	rep(i,l,r)
		ek[qu0[i].cur].w=inf;
	kruscal(poi0,cnt0);
	totk=tot0;
	rep(i,l,r)flg[qu0[i].cur]=1;
	rep(i,1,tot0){
		ek[i]=(EDGE2){e0[i].a,e0[i].b,e0[i].w,i};
		if(!flg[i])e0[i].a=0;
	}
	rep(i,l,r)ek[qu0[i].cur].w=-inf;
	kruscal(poi0,cnt0);
	EDGE *e1=Hz[dep+1].e;QUERY *qu1=Hz[dep+1].qu;
	int &tot1=Hz[dep+1].tot,*poi1=Hz[dep+1].poi,&cnt1=Hz[dep+1].cnt;
	int x,mid=l+r>>1;
	rep(i,l,r)flg[qu0[i].cur]=0;
	rep(i,1,tot0){
		if(!e0[i].a)continue;
		if(flg[i]){
			cmb(par0,sz0,e0[i].a,e0[i].b);
			e0[i].a=0;sum+=e0[i].w;
		}
	}
	cnt1=tot1=0;
	rep(i,1,cnt0){
		x=poi0[i];
		if(getpar(par0,x)==x)poi1[++cnt1]=x;
	}
	rep(i,1,tot0)if(e0[i].a)
		e1[key[i]=++tot1]=(EDGE){getpar(par0,e0[i].a),getpar(par0,e0[i].b),e0[i].w};
	rep(i,l,r)qu1[i]=(QUERY){key[qu0[i].cur],qu0[i].w};
	cdq(l,mid,sum,dep+1);
	tot1=0;
	rep(i,1,tot0)if(e0[i].a)
		e1[++tot1]=(EDGE){getpar(par0,e0[i].a),getpar(par0,e0[i].b),e0[i].w};
	rep(i,l,mid)e1[qu1[i].cur].w=qu1[i].w;
	cdq(mid+1,r,sum,dep+1);
}
int main(){
	int i;
	read(n);read(m);read(Q);
	EDGE *e0=Hz[1].e;QUERY *qu0=Hz[1].qu;
	rep(i,1,m){
		read(e0[i].a);read(e0[i].b);read(e0[i].w);
	}
	Hz[1].tot=m;
	int *poi0=Hz[1].poi;
	rep(i,1,n)poi0[i]=i;
	Hz[1].cnt=n;
	rep(i,1,Q){
		read(qu0[i].cur);read(qu0[i].w);
	}
	cdq(1,Q,0LL,1);
	rep(i,1,Q)printf("%lld\n",ans[i]);
	return 0;
}