【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;
}
用先进的科学技术与思想文化武装起来的人民,是一个国家最巨大的生产力,是一个社会最强大的推动力,是一个民族最坚实的自信力,是一支军队最可靠的战斗力。
——A·H