Codeforces-707D-Persistent Bookcase（主席树 单点修改 区间求和）

题目链接

题意

在书架上放书，有四种操作。

1： 在（i , j）位置放书，如果有书就不放；

2：在（i，j）位置拿走书，如果没有书就不拿；

3：在第 i 个书柜，求异或，有书的变为无书，无书的变为有书；

4：回到第k次修改的样子。

1、2、3操作都是单点修改，可以用线段树维护。第4个操作需要前面线段树的信息，所以要建可持续化线段树。

在线做法：主席树，O(nlogn);

离线做法：DFS，遇到操作4回溯，O(n)。

 

主席树数组大小应该开 n * log q， q是修改的次数。

---

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5+10;

struct Tree
{
    int l,r,sum;
    bitset <1010> bt;
}tr[N*11];

bitset <1010> base;
int m,n,q,op,x,y;
int root[N],cnt = 1;

void update(int &k,int l,int r,int op,int x,int y)
{
    tr[cnt] = tr[k];
    k = cnt++;
    if(l==r){
        if(op==1) tr[k].bt.set(y);
        else if(op==2) tr[k].bt.reset(y);
        else if(op==3) tr[k].bt ^= base;
        tr[k].sum = tr[k].bt.count();
        return;
    }
    int mid = (l+r)/2;
    if(x<=mid) update(tr[k].l,l,mid,op,x,y);
    else update(tr[k].r,mid+1,r,op,x,y);
    tr[k].sum = tr[tr[k].l].sum + tr[tr[k].r].sum;
}

int main()
{
    scanf("%d%d%d",&n,&m,&q);
    for(int i=1;i<=m;i++) base.set(i);
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d",&op,&x);
        if(op==1||op==2) scanf("%d",&y);
        if(op==4){
            root[i] = root[x];
        }
        else{
            root[i] = root[i-1];
            update(root[i],1,n,op,x,y);
        }
        printf("%d\n",tr[ root[i] ].sum);
    }

    return 0;
}