I Like Matrix Forever!【搜索树】

>Description
对一个 n ∗ m 的零矩阵 A 进行 q 次操作：
• 1 i j：将 Ai,j 取反；
• 2 i：将矩阵 A 第 i 行的所有元素全部取反；
• 3 j：将矩阵 A 第 j 列的所有元素全部取反；
• 4 k：将矩阵 A 还原为第 k 次操作之后的状态。
进行每一次操作之后，询问当前矩阵所有元素的和。

---

>Input
第一行包含三个整数 n，m 和 q。
之后 q 行每行包含两个或三个整数，表示一次操作的所有参数。

>Output
共 q 行每行包含一个整数 ans，表示当前矩阵所有元素的和。

---

>Sample Input
2 2 4
2 1
1 1 2
4 1
3 2

>Sample Output
2
1
2
2

对于 30% 的数据：n, m ≤ 300，q ≤ 1000
对于 60% 的数据：n, m ≤ 1000，q ≤ 5000
对于 100% 的数据：n, m ≤ 1000，q ≤ 100000

---

>解题思路
搜索树：把全部的数据的序号按照树的结构一直连下去。遇到输入的是操作4（需要返回到操作x[i]）的话，就把i连接到x[i]，其余的直接连接到i-1就行了，接下来搜索树形结构。

然后还有使用快读和快输。

---

>代码

#include<iostream>
#include<cstdio>
using namespace std;
struct ooo
{
	int to,next;
}f[100005];
int n,m,q,tt,sum,h[100005],t[100005],x[100005],y[100005],ans[100005];
bool a[1005][1005];
void lil(int s)
{
	if(t[s]==1)
	{
		a[x[s]][y[s]]=!a[x[s]][y[s]];
		if(a[x[s]][y[s]]) sum++;
		else sum--;
	}
	else if(t[s]==2)
	{
		for(int j=1;j<=m;j++)
		{
			a[x[s]][j]=!a[x[s]][j];
			if(a[x[s]][j]) sum++;
			else sum--;
		}
	}
	else if(t[s]==3)
	{
		for(int j=1;j<=n;j++)
		{
			a[j][x[s]]=!a[j][x[s]];
			if(a[j][x[s]]) sum++;
			else sum--;
		}
	} //操作
	
	ans[s]=sum; //记录数据
	for(int i=h[s];i;i=f[i].next)
	 lil(f[i].to); //搜索树
	 
	if(t[s]==1)
	{
		a[x[s]][y[s]]=!a[x[s]][y[s]];
		if(a[x[s]][y[s]]) sum++;
		else sum--;
	}
	else if(t[s]==2)
	{
		for(int j=1;j<=m;j++)
		{
			a[x[s]][j]=!a[x[s]][j];
			if(a[x[s]][j]) sum++;
			else sum--;
		}
	}
	else if(t[s]==3)
	{
		for(int j=1;j<=n;j++)
		{
			a[j][x[s]]=!a[j][x[s]];
			if(a[j][x[s]]) sum++;
			else sum--;
		}
	} //回溯
}
int input()
{
	char c=getchar();
	while(c<'0'||c>'9') c=getchar();
	int l=0;
	while(c>='0'&&c<='9')
	{
		l=l*10+c-'0';
		c=getchar();
	}
	return l;
} //快读
int output(int l)
{
	if(l>9) output(l/10);
	putchar(l%10+'0');
} //快输
int main()
{
	n=input(); m=input(); q=input();
	for(int i=1;i<=q;i++)
	{
		t[i]=input();
		if(t[i]==1) x[i]=input(),y[i]=input();
		else x[i]=input(); //不同的操作是输入不同的数的个数
		if(t[i]==4) f[++tt].to=i,f[tt].next=h[x[i]],h[x[i]]=tt;
		else f[++tt].to=i,f[tt].next=h[i-1],h[i-1]=tt; //连接
	}
	lil(0); //一定要从0版本开始搜，防止有的数据是返回第0次操作
	for(int i=1;i<=q;i++)
	 output(ans[i]),putchar(10); //输出
	return 0;
}