bzoj 3109: [cqoi2013]新数独【dfs】

按3x3的小块dfs，填数的时候直接满足所有条件即可

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=15;
int a[N][N],ans[N][N];
char b[N][N],c[N][N];
bool v[5][5][N],h[N][N],l[N][N];
char read()
{
	char p=getchar();
	while(p!='^'&&p!='v'&&p!='<'&&p!='>')
		p=getchar();//cerr<<p<<endl;
	return p;
}
bool ok(int x,int y,int i)
{//cerr<<b[x][y-1]<<" "<<x<<" "<<y<<" "<<i<<"   "<<a[x][y-1]<<endl;
	if(h[x][i]||l[y][i])
		return 0;
	if(y%3!=1)
	{
		if(b[x][y-1]=='>'&&a[x][y-1]<i)
			return 0;
		if(b[x][y-1]=='<'&&a[x][y-1]>i)
			return 0;
	}
	if(x%3!=1)
	{
		if(c[x-1][y]=='^'&&a[x-1][y]>i)
			return 0;
		if(c[x-1][y]=='v'&&a[x-1][y]<i)
			return 0;
	}
	return 1;
}
void dfs(int p,int q,int x,int y)
{
	if(ans[1][1])
		return;
	if(p==3)
	{
		for(int i=1;i<=9;i++)
			for(int j=1;j<=9;j++)
				ans[i][j]=a[i][j];
		return;
	}
	for(int i=1;i<=9;i++)
		if(!v[p][q][i]&&ok(p*3+x,q*3+y,i))
		{
			v[p][q][i]=1,h[p*3+x][i]=1,l[q*3+y][i]=1,a[p*3+x][q*3+y]=i;
			if(q==2&&x==3&&y==3)
				dfs(p+1,0,1,1);
			else if(x==3&&y==3)
				dfs(p,q+1,1,1);
			else if(y==3)
				dfs(p,q,x+1,1);
			else
				dfs(p,q,x,y+1);
			v[p][q][i]=0,h[p*3+x][i]=0,l[q*3+y][i]=0,a[p*3+x][q*3+y]=0;
		}
}
int main()
{
	for(int i=0;i<3;i++)
	{
		for(int j=0;j<6;j++)
			b[i*3+1][j/2*3+j%2+1]=read();
		for(int j=1;j<=9;j++)
			c[i*3+1][j]=read();
		for(int j=0;j<6;j++)
			b[i*3+2][j/2*3+j%2+1]=read();
		for(int j=1;j<=9;j++)
			c[i*3+2][j]=read();
		for(int j=0;j<6;j++)
			b[i*3+3][j/2*3+j%2+1]=read();
	}
	// for(int i=1;i<=9;i++)
	// {
		// for(int j=1;j<=9;j++)
			// cerr<<c[i][j];
		// cerr<<endl;
	// }
	dfs(0,0,1,1);
	for(int i=1;i<=9;i++)
	{
		for(int j=1;j<=9;j++)
			printf("%d ",ans[i][j]);
		puts("");
	}
	return 0;
}