【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;
}

转载于:https://www.cnblogs.com/AH2002/p/10524089.html