Codeforces Round #368 (Div. 2) D. Persistent Bookcase(n层m个的书架的四种操作)

传送门:Codeforces Round #368 (Div. 2) D. Persistent Bookcase

---

题意:给你一个有n层,每个有m个的书架,有四种操作 (1 ≤ n, m ≤ 10^3, 1 ≤ q ≤ 10^5)
1 i j 表示在第i层j位置放入一本书(原来不存在)
2 i j表示在第i层j位置拿出一本书(原来存在)
3 i 表示将第i层的每个位置的书相反一下,原来不存在,现在存在
4 i 表示回到第i次操作的状态
对于每一次查询,输出还剩下多少本书

---

思路:我们可以根据操作建立一棵操作树,建立完操作树之后就只有前3种操作了,我们可以进行回溯的dfs,前面三种操作利用bitset进行维护就可以了

#include<bits/stdc++.h>
using namespace std;
const int maxn=100100;
bitset<1001>b[1001],base;
vector<int>G[maxn];
int ans=0,n,m,ANS[maxn];
struct query{
    int op,u,v;
}Q[maxn];

void change(int u){
    int num=b[Q[u].u].count();
    ans=ans-num+(m-num);
    b[Q[u].u].flip();
    b[Q[u].u]&=base;
}

void dfs(int u){
    int num=-1;
    if(u!=0&&Q[u].op!=4){
        if(Q[u].op==1){
            if(b[Q[u].u][Q[u].v]==0)
                ans++,num=0,b[Q[u].u][Q[u].v]=1;
        }
        else if(Q[u].op==2){
            if(b[Q[u].u][Q[u].v]==1)
                ans--,num=1,b[Q[u].u][Q[u].v]=0;
        }
        else
            change(u);
    }
    ANS[u]=ans;
    for(int i=0;i<G[u].size();i++)
        dfs(G[u][i]);
    if(u!=0&&Q[u].op!=4){
        if(Q[u].op==1){
            if(num==0)
                ans--,b[Q[u].u][Q[u].v]=0;
        }
        else if(Q[u].op==2){
            if(num==1)
                ans++,b[Q[u].u][Q[u].v]=1;
        }
        else
            change(u);
    }
}

int main(){
    int q;
    scanf("%d%d%d",&n,&m,&q);
    for(int i=0;i<m;i++)
        base[i]=1;
    for(int i=1;i<=q;i++){
        scanf("%d",&Q[i].op);
        if(Q[i].op==1||Q[i].op==2)
            scanf("%d%d",&Q[i].u,&Q[i].v),Q[i].v--,G[i-1].push_back(i);
        else{
            scanf("%d",&Q[i].u);
            if(Q[i].op==4)
                G[Q[i].u].push_back(i);
            else
                G[i-1].push_back(i);
        }
    }
    dfs(0);
    for(int i=1;i<=q;i++)
        printf("%d\n",ANS[i]);
    return 0;
}

主席树强制在线化

#include<bits/stdc++.h>
using namespace std;
const int maxn=100001;
struct node{
    int l,r,w;//L,R表示覆盖的区间
    bitset<1000>b;
}T[maxn*11];
int sz=0,root[maxn],op,m;
bitset<1000>base;

void update(int &i,int l,int r,int pos,int cnt){
    T[++sz]=T[i],i=sz;
    if(l==r){
        if(op==1){
            if(T[i].b[cnt]==0)
                T[i].w++,T[i].b[cnt]=1;
        }
        else if(op==2){
            if(T[i].b[cnt]==1)
                T[i].w--,T[i].b[cnt]=0;
        }
        else{
            int num=T[i].b.count();
            T[i].w=T[i].w-num+(m-num);
            T[i].b.flip();
            T[i].b&=base;
        }
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)
        update(T[i].l,l,mid,pos,cnt);
    else
        update(T[i].r,mid+1,r,pos,cnt);
    T[i].w=T[T[i].l].w+T[T[i].r].w;
}

int main(){
    int n,q,u,v;
    scanf("%d%d%d",&n,&m,&q);
    for(int i=0;i<m;i++)
        base[i]=1;
    root[0]=0;
    for(int i=1;i<=q;i++){
        scanf("%d",&op);
        if(op==1||op==2){
            scanf("%d%d",&u,&v),v--;
            root[i]=root[i-1];
            update(root[i],1,n,u,v);
        }
        else{
            scanf("%d",&u);
            if(op==4)
                root[i]=root[u];
            else
                root[i]=root[i-1],update(root[i],1,n,u,0);
        }
        printf("%d\n",T[root[i]].w);
    }
    return 0;
}