Codeforces 707D.Persistent Bookcase（离线算法，dfs树，好题！）_codeforces

本文链接：https://blog.csdn.net/cmershen/article/details/52396598

本文介绍了一种使用模拟与图遍历的方法来解决CodeForces平台上的707D题目。该题涉及操作书架的书籍放置、移除、翻转等，并需返回每步操作后书的数量。通过构建操作树并利用DFS进行遍历更新书的数量，最终输出每次操作后的书本总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门http://codeforces.com/problemset/problem/707/D
题目大意：
有一个书架，书架有n层（分别编号为1…n），每层能放m本书（分别编号为1…m），有q次操作(分别编号为1…q）：
1 i j：在第i层第j位放一本书（如果这个位置没有书）
2 i j ：在第i层第j位取走一本书（如果这个位置有书）
3 i:翻转第i行（无书变有书，有书变无书）
4 k:回退到第k个操作后的状态，若k=0则为清空
输出每次操作后书架上有几本书。
题目分析：
一开始没什么思路，以为是线段树。去百度搜了一下大神的做法，想了大半个下午终于想明白了。
用布尔型二维数组bool book[1005][1005]模拟书架，用一棵树记录每个操作之后的状态，树的节点记录下操作号(op,i,j)和此时书架上书的数目(ans)。读入数据时记录树的邻接表，其中遇到1，2，3操作依次向子节点延伸，遇到4操作在回退的操作编号上增加一个子节点，然后dfs遍历这棵树，更新每个节点的数目域。
这样说可能有点不太明白，看下面的测试用例：

首先树根的数目域为0，接下来按输入的操作顺序建树如下：

测试用例对应的树型结构

然后dfs这个树，读每个节点的操作号，如果是1，则子节点的ans可能+1也可能不变，如果是2则-1或不变，如果是3则统计第i行中有书的个数，用m减这个个数。如果是4则直接向下搜索。（例如节点9和节点12的ans域是一样的，因为12的操作为4 9）。记得向下搜索完回溯，保证搜索到第i个节点时，书架的情况正好是第i次操作后的书架。

本题似乎还可以用二维线段树解决？不过操作4怎么办？

AC代码：

#include <bits/stdc++.h>
using namespace std;
struct node {
  int op,i,j;//每次操作的操作数
  int ans;//当前节点中书的数目
};
node p[100005];
bool book[1005][1005];
vector<int> g[100005];//邻接表
int n,m,q;
void dfs(int t,int cur) {
  p[t].ans=cur;
  for(int i=0;i<g[t].size();i++) {
    int next=g[t][i];
    if(p[next].op==1) {
      if(book[p[next].i][p[next].j]) {//操作1下如果有书则书架不变
        dfs(next,cur);
      }
      else {
        book[p[next].i][p[next].j]=true;
        dfs(next,cur+1);
        book[p[next].i][p[next].j]=false;//回溯，保持书架情况与当前节点一致
      }
    }
    else if(p[next].op==2) {
      if(!book[p[next].i][p[next].j]) {
        dfs(next,cur);
      }
      else {
        book[p[next].i][p[next].j]=false;
        dfs(next,cur-1);
        book[p[next].i][p[next].j]=true;
      }
    }
    else if(p[next].op==3) {
      int change=0;//要改变的书本数目
      for(int j=1;j<=m;j++) {
        if(book[p[next].i][j]) {
          book[p[next].i][j]=false;
          change--;
        }
        else {
          book[p[next].i][j]=true;
          change++;
        }
      }
      dfs(next,cur+change);
      for(int j=1;j<=m;j++)
        book[p[next].i][j]=!book[p[next].i][j];
    }
    else {
      dfs(next,cur); //跳转不改变书本数量
    }
  }
}
int main() {
  memset(book,false,sizeof(book));
  p[0].ans=0;
  scanf("%d %d %d", &n,&m,&q);
  for(int i=1;i<=q;i++) {
    scanf("%d", &p[i].op);
    if(p[i].op==1 || p[i].op==2) {
      g[i-1].push_back(i);//向下顺序建树
      scanf("%d %d", &p[i].i,&p[i].j);
    }
    else if(p[i].op==3) {
      g[i-1].push_back(i);
      scanf("%d", &p[i].i);
    }
    else {
      scanf("%d", &p[i].i);
      g[p[i].i].push_back(i);//要退回的节点增加一个孩子
    }
  }
  dfs(0,0);//0号节点为根节点，有0本书
  for(int i=1;i<=q;i++)
    printf("%d\n", p[i].ans);
}