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