51nod 2615 修改01矩阵(递归枚举)

有一个n*n的01矩阵，每次操作可以选择一个位置改变这个位置元素的值(0变成1，1变成0)，但是这个位置的上下左右四个位置的元素的状态也会发生改变(0变成1，1变成0),问是否可以通过这个操作，将01矩阵所有元素都变成0，操作次数不限，也可以为0次。

1 1 1
1 0 1
1 1 1

在这个数据中，可以使所有元素都变成0的操作方式为：

1 1 1
1 0 1
1 1 1

1代表对这个位置的元素进行一次操作，0代表没有进行。

输入
第1行：一个整数n表示矩阵的大小为n*n。(1<=n<=10)
第2~n+1行，每行n个正整数，第i行第j个正整数表示01矩阵第i行第j列位置元素的值a[i][j]。(0<=a[i][j]<=1)
输出
如果可以通过上述操作将01矩阵的所有元素变为0，输出"Yes",否则输出"No"。
输入样例
3
1 1 1
1 0 1
1 1 1
输出样例
Yes

只递归枚举第一行即可！
其他行按上一行的结果递推即可：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;

int n;
int a[12][12];
int dir[5][2]={0,0,0,1,0,-1,1,0,-1,0};
int flag=0;

bool check(){
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			if(a[i][j]!=0){
				return false;
			}
		}
	}
	return true;
}

void flip(int x,int y){
	int x1,y1;
	for(int i=0;i<=4;i++){
		x1=x+dir[i][0];
		y1=y+dir[i][1];
		if(x1>=0&&x1<n&&y1>=0&&y1<n){
			a[x1][y1]=1-a[x1][y1];
		}
	}
}

void dfs(int x,int y){
	if(check()||flag){
		flag=1;
		return;
	}
	if(x>=n||y>=n) return;
	int x1,y1;
	
	x1=x+(y+1)/n;
	y1=(y+1)%n;
	if(a[x-1][y]==1){
		flip(x,y);
		dfs(x1,y1);
	}else{
		dfs(x1,y1);
	}
}

void firstrow(int col){
	if(col==n){
		dfs(1,0);
		return;
	}
	flip(0,col);
	firstrow(col+1);
	flip(0,col);
	firstrow(col+1);
}

int main() {
	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			cin>>a[i][j];
		}
	}
	
	firstrow(0);
	if(flag){
		cout<<"Yes"<<endl;
	}else{
		cout<<"No"<<endl;
	}
	return 0;
}