【CDQ分治】[HNOI2010]城市建设

题目链接

线段树分治+LCT只有80

然后就有了CDQ分治的做法

把不可能在生成树里的扔到后面

把一定在生成树里的扔到并查集里存起来

分治到l=r,修改边权,跑个kruskal就行了

由于要支持撤销,并查集要按秩合并

#include"cstdio"
#include"cstring"
#include"iostream"
#include"algorithm"
using namespace std;

const int MAXN=2e4+5;
const int MAXM=5e4+5;
const int INF=1e9;

int n,m,q,t,ct,p;
int f[MAXN],mxdp[MAXN];
long long sum;
long long ans[MAXM];
bool vis[MAXM];
int v[MAXM];
struct rpg{
	int ls,nx,ln,id,kd;
}a[MAXM],st[MAXM];
struct vx{
	int id,val;
}w[MAXM];
struct fx{
	int x,fx,kd;
}stk[MAXM];

int read()
{
	int x=0;char ch=getchar();
	while(ch<'0'||'9'<ch) ch=getchar();
	while('0'<=ch&&ch<='9') x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return x;
}

int find(int x){return f[x]==x?x:find(f[x]);}

bool cmp1(rpg a,rpg b){return a.ln<b.ln;}
bool cmp2(rpg a,rpg b){return a.kd<b.kd;}
void un(int fa,int fb)
{
	if(mxdp[fa]>mxdp[fb]) swap(fa,fb);
	if(mxdp[fa]==mxdp[fb]) stk[++ct]=(fx){fa,fb,1},++mxdp[fb];
	else stk[++ct]=(fx){fa,fb,0};
	f[fa]=fb;
}

long long calc(int d,int t2)
{
	for(int i=1;i<=t;++i){
		if(a[i].id==w[d].id){
			st[w[d].id].ln=a[i].ln=w[d].val;
			break;
		}
	}sort(a+1,a+t+1,cmp1);
	long long s=0;
	for(int i=1;i<=t;++i){
		int fa=find(a[i].ls),fb=find(a[i].nx);
		if(fa!=fb) un(fa,fb),s+=a[i].ln;
	}while(ct>t2){
		if(stk[ct].kd) --mxdp[stk[ct].fx];
		f[stk[ct].x]=stk[ct].x;
		--ct;
	}return sum+s;
}

void C(int t2)
{
	for(int i=1;i<=t;++i) if(vis[a[i].id]) a[i].ln=INF;
	sort(a+1,a+t+1,cmp1);
	for(int i=1;i<=t&&a[i].ln<INF;++i){
		int fa=find(a[i].ls),fb=find(a[i].nx);
		if(fa!=fb) un(fa,fb);
		else a[i].kd=2;
	}sort(a+1,a+t+1,cmp2);
	while(t&&a[t].kd) --t;
	while(ct>t2){
		if(stk[ct].kd) --mxdp[stk[ct].fx];
		f[stk[ct].x]=stk[ct].x;
		--ct;
	}return;
}

void D(int t2)
{
	for(int i=1;i<=t;++i) if(vis[a[i].id]) a[i].ln=-INF;
	sort(a+1,a+t+1,cmp1);
	for(int i=1;i<=t;++i){
		int fa=find(a[i].ls),fb=find(a[i].nx);
		if(fa!=fb){
			un(fa,fb);
			if(a[i].ln>-INF) a[i].kd=1,sum+=a[i].ln;
		}
	}sort(a+1,a+t+1,cmp2);
	while(t&&a[t].kd) --t;
	while(ct>t2){
		if(stk[ct].kd) --mxdp[stk[ct].fx];
		f[stk[ct].x]=stk[ct].x;
		--ct;
	}return;
}

void reD(int t2)
{
	while(ct>t2){
		if(stk[ct].kd) --mxdp[stk[ct].fx];
		f[stk[ct].x]=stk[ct].x;
		--ct;
	}return;
}

void reC(int t1)
{
	for(int i=t+1;i<=t1;++i){
		if(a[i].kd==1) sum-=a[i].ln;
		a[i].kd=0;
	}t=t1;
	return;
}

void CDQ(int l,int r)
{
	if(l==r){ans[l]=calc(l,ct);return;}
	int ctp=t,cmd=ct,mid=l+r>>1;
	for(int i=1;i<=t;++i) vis[a[i].id]=0;
	for(int i=l;i<=r;++i) vis[w[i].id]=1;
	C(cmd),D(cmd);
	for(int i=t+1;i<=ctp;++i) if(a[i].kd==1) un(find(a[i].ls),find(a[i].nx));
	for(int i=1;i<=t;++i) a[i].ln=st[a[i].id].ln;
	CDQ(l,mid),CDQ(mid+1,r);
	reC(ctp);reD(cmd);
	return;
}

int main()
{
	n=read(),t=m=read(),q=read();
	for(int i=1;i<=n;++i) f[i]=i;
	for(int i=1;i<=m;++i) a[i].ls=read(),a[i].nx=read(),a[i].ln=read(),a[i].id=i,st[i]=a[i];
	for(int i=1;i<=q;++i) w[i].id=read(),w[i].val=read();
	CDQ(1,q);for(int i=1;i<=q;++i) printf("%lld\n",ans[i]);
	return 0;
}
posted @ 2019-03-13 16:35  A·H  阅读(222)  评论(0编辑  收藏  举报