Codeforces Round #368 (Div. 2) D. Persistent Bookcase 离线 dfs+bitset

参考http://www.cnblogs.com/qscqesze/p/5791698.html

题意

你有n个书架，每个书架有m层。

有四个操作

1 i j，在第i个书架第j层放一本书。

2 i j，把第i个书架第j层的书扔掉

3 i，把第三层的所有书的状态取反，就有的变没，没的变有

4 k，回到第k个询问时候的状态。

先看不到样例。wa13.。

int n,m,q;
bitset<1010>mask;
bitset<1010>bit[N];
vector<int>G[N*100];
int ans[N*100];
int a[N*100],b[N*100];
int op[N*100];
void dfs(int x){
    if(x==0){
        for(int i=0;i<G[x].size();++i){
            ans[G[x][i]]=ans[x];
            dfs(G[x][i]);
        }
    }
    int f=0;
    if(op[x]==1){
        if(bit[a[x]][b[x]]==0){
            f=1;
            bit[a[x]][b[x]]=1;
            ans[x]++;
        }
        for(int i=0;i<G[x].size();++i){
            ans[G[x][i]]=ans[x];
            dfs(G[x][i]);
        }
        if(f)bit[a[x]][b[x]]=1;
    }
    if(op[x]==2){
        if(bit[a[x]][b[x]]==1){
            bit[a[x]][b[x]]=0;
            ans[x]--;
            f=1;
        }
        for(int i=0;i<G[x].size();++i){
            ans[G[x][i]]=ans[x];
            dfs(G[x][i]);
        }
        if(f)bit[a[x]][b[x]]=0;
    }
    if(op[x]==3){
        ans[x]-=bit[a[x]].count();
        bit[a[x]]^=mask;
        ans[x]+=bit[a[x]].count();
        for(int i=0;i<G[x].size();++i){
            ans[G[x][i]]=ans[x];
            dfs(G[x][i]);
        }
        bit[a[x]]^=mask;
    }
    if(op[x]==4){
        for(int i=0;i<G[x].size();++i){
            ans[G[x][i]]=ans[x];
            dfs(G[x][i]);
        }
    }
}

int main(){
    sf("%d%d%d",&n,&m,&q);
    for(int i=1;i<=m;++i)mask[i]=1;
    for(int i=1;i<=q;++i){
        sf("%d",&op[i]);
        if(op[i]==1){ sf("%d%d",&a[i],&b[i]); }
        if(op[i]==2){ sf("%d%d",&a[i],&b[i]); }
        if(op[i]==3){ sf("%d",&a[i]); }
        if(op[i]==4){
            sf("%d",&a[i]);
            G[a[i]].push_back(i);
        }
        else{ G[i-1].push_back(i); }
    }
    dfs(0);
    for(int i=1;i<=q;++i) pf("%d\n",ans[i]);
}