Description 一个n*n的数独,每一行每一列数字都不同,一个数组表示该位置放这么高的塔,已经填好一些数字并且给出一些限制,如某一行左边的限制为从左边看这一行能看到多少塔,上下左右都可能有类似的限制,问是否存在一个合法的数独满足这些条件,如果有则输出一组合法解,否则输出no Input 第一行一整数T表示用例组数,每组用例首先输入一整数n表示数独大小,然后是一个(n+2)*(n+2)的矩阵,第一行和最后一列分别表示上面和下面的限制,第一列和最后一列分别表示左边和右边的限制,-表示没有限制,中间的n*n矩阵表示数独初始状态,-表示没填数(3<=n<=5) Output 如果存在一个合法的数独满足这些条件则输出一组合法解,否则输出no Sample Input Sample Output Solution 暴力dfs,dfs的时候先不管四周的限制,只根据数独的限制去放数,等所有数字都放完再判断其是否满足周围的限制 Code
#include<cstdio> #include<iostream> #include<cstring> #include<algorithm> #include<cmath> #include<vector> #include<queue> #include<map> #include<set> #include<ctime> using namespace std; typedef long long ll; #define INF 0x3f3f3f3f #define maxn 7 int T,n,flag,mark[maxn],a[maxn][maxn]; char s[maxn][maxn]; bool check1() { for(int i=1;i<=n;i++) { memset(mark,0,sizeof(mark)); for(int j=1;j<=n;j++) if(a[i][j]) { if(mark[a[i][j]])return 0; mark[a[i][j]]=1; } } for(int j=1;j<=n;j++) { memset(mark,0,sizeof(mark)); for(int i=1;i<=n;i++) if(a[i][j]) { if(mark[a[i][j]])return 0; mark[a[i][j]]=1; } } return 1; } bool check2() { for(int i=1;i<=n;i++) { int temp,num; if(s[i][0]!='-') { temp=num=0; for(int j=1;j<=n;j++) if(a[i][j]>temp)temp=a[i][j],num++; if(num!=s[i][0]-'0')return 0; } if(s[i][n+1]!='-') { temp=num=0; for(int j=n;j>=1;j--) if(a[i][j]>temp)temp=a[i][j],num++; if(num!=s[i][n+1]-'0')return 0; } } for(int j=1;j<=n;j++) { int temp,num; if(s[0][j]!='-') { temp=num=0; for(int i=1;i<=n;i++) if(a[i][j]>temp)temp=a[i][j],num++; if(num!=s[0][j]-'0')return 0; } if(s[n+1][j]!='-') { temp=0,num=0; for(int i=n;i>=1;i--) if(a[i][j]>temp)temp=a[i][j],num++; if(num!=s[n+1][j]-'0')return 0; } } return 1; } int dfs(int x,int y) { if(flag)return 1; if(y==n+1)x++,y=1; if(x==n+1) { if(check2()==1) { flag=1; return 1; } return 0; } if(a[x][y])return dfs(x,y+1); int gg=1; for(int i=1;i<=n;i++) { a[x][y]=i; if(check1()) { if(dfs(x,y+1)) { gg=0; flag=1; return 1; } } a[x][y]=0; } if(gg)return 0; } int main() { scanf("%d",&T); while(T--) { scanf("%d",&n); for(int i=0;i<=n+1;i++)scanf("%s",s[i]); memset(a,0,sizeof(a)); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) if(s[i][j]>='0'&&s[i][j]<='9') a[i][j]=s[i][j]-'0'; flag=0; dfs(1,1); if(flag) { for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++)printf("%d",a[i][j]); printf("\n"); } } else printf("no\n"); printf("\n"); } return 0; }