Codeforces 707D.Persistent Bookcase(离线算法,dfs树,好题!)

来源:互联网 发布:生命最后的读书会 知乎 编辑:程序博客网 时间:2024/06/05 15:25

传送门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遍历这棵树,更新每个节点的数目域。
这样说可能有点不太明白,看下面的测试用例:

19 16 143 103 83 32 5 31 7 154 02 11 21 16 31 16 33 131 13 34 91 5 113 1

首先树根的数目域为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);}
0 0