【Codeforces 707 D】离线有向树

CF707D
题意 给你n个书架 每个书架上有m个位置可以放书
1 x y 操作代表 如果 x 书架上 y 位置是空的 那么就放一本书进去
2 x y 操作代表 如果 x 书架上 y 位置是有书的 那么就拿出一本书
3 x 操作代表 给 x 暑假取反 有书的部分取出书 没书的部分放入书
4 x 操作 代表 x 操作后的那个状态
对于每个操作执行结束后 都要输出书架上当前有多少书
然后对于这种问题 因为有4操作 所以不能简单的建线段树来实现
跟博主学习了一下 是把操作当做节点 建一颗有向的操作树
因为 n 和 m 都只有 1e3 那么大 所以暴力修改就好了
也就是这颗树在你实现了那些操作以后 必须回溯回去
但是要注意一点 如果书架上本来就有书了 那么你就不能放进去
所以我们开了一个vis代表能不能实现那步操作
然后 4 操作也就是在 x 那个节点插入那个 i

/*
    if you can't see the repay
    Why not just work step by step
    rubbish is relaxed
    to ljq
*/
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <cmath>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#include <vector>
#include <stdlib.h>
#include <algorithm>
using namespace std;

#define dbg(x) cout<<#x<<" = "<< (x)<< endl
#define dbg2(x1,x2) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<endl
#define dbg3(x1,x2,x3) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<" "<<#x3<<" = "<<x3<<endl
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
#define lc (rt<<1)
#define rc (rt<<11)
#define mid ((l+r)>>1)

typedef pair<int,int> pll;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int _inf = 0xc0c0c0c0;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll _INF = 0xc0c0c0c0c0c0c0c0;
const ll mod =  (int)1e9+7;

ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll ksm(ll a,ll b,ll mod){int ans=1;while(b){if(b&1) ans=(ans*a)%mod;a=(a*a)%mod;b>>=1;}return ans;}
ll inv2(ll a,ll mod){return ksm(a,mod-2,mod);}
void exgcd(ll a,ll b,ll &x,ll &y,ll &d){if(!b) {d = a;x = 1;y=0;}else{exgcd(b,a%b,y,x,d);y-=x*(a/b);}}//printf("%lld*a + %lld*b = %lld\n", x, y, d);
const int MAX_N = 1025;
int x[100025],opt[100025],y[100025],sum,ans[100025];
bool vis[100025],book[1025][1025];
vector<int > G[100025];
int n,m;
void dfs(int u)
{
    if(opt[u]==1)
    {
        if(!book[x[u]][y[u]]) vis[u] = true,book[x[u]][y[u]] = true,sum++;
        ans[u] = sum;
    }
    else if(opt[u]==2)
    {
        if(book[x[u]][y[u]]) vis[u] = true,book[x[u]][y[u]] = false,sum--;
        ans[u] = sum;
    }
    else if(opt[u]==3)
    {
        for(int i = 1;i<=m;++i)
            if(book[x[u]][i]) book[x[u]][i] = false,sum--;
            else book[x[u]][i] = true,sum++;
        ans[u] = sum;
    }
    else ans[u] = ans[x[u]];

    for(int i = 0;i<G[u].size();++i) dfs(G[u][i]);

    if(opt[u]==1)
    {
        if(book[x[u]][y[u]]&&vis[u]) book[x[u]][y[u]] = false,sum--;
    }
    else if(opt[u]==2)
    {
        if(!book[x[u]][y[u]]&&vis[u]) book[x[u]][y[u]] = true,sum++;
    }
    else if(opt[u]==3)
    {
        for(int i = 1;i<=m;++i)
            if(!book[x[u]][i]) book[x[u]][i] = true,sum++;
            else book[x[u]][i] = false,sum--;
    }
}
int main()
{
    //ios::sync_with_stdio(false);
    //freopen("a.txt","r",stdin);
    //freopen("b.txt","w",stdout);
    int Q;scanf("%d%d%d",&n,&m,&Q);
    for(int i = 1;i<=Q;++i)
    {
        scanf("%d%d",&opt[i],&x[i]);
        if(opt[i]<3) scanf("%d",&y[i]);
        if(opt[i]==4) G[x[i]].push_back(i);
        else G[i-1].push_back(i);
    }
    dfs(0);
    for(int i = 1;i<=Q;++i) printf("%d\n",ans[i]);
    //fclose(stdin);
    //fclose(stdout);
    //cout << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}