传送门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 14
3 10
3 8
3 3
2 5 3
1 7 15
4 0
2 11 2
1 16 3
1 16 3
3 13
1 13 3
4 9
1 5 11
3 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);
}