递归——简单数谜

问题 e: 【递归入门】简单数谜

时间限制: 1 Sec  内存限制: 128 MB
提交: 20  解决: 10
[提交][状态][讨论版][命题人:外部导入]

题目描述

很多人都曾经听说过数独，但你是否听说过数谜（Karuro）呢？实际上，数谜是数独的更大（且更难）的兄弟问题，而且在日本也是非常受欢迎的。 
　　数谜问题和填字游戏类似，不过它要填的不是文字而是数字。数谜游戏的目标是用1-9填满所有空格，且这些数字相加的和满足相应的要求（或者称为“提示”），且在同一栏（“栏”是指一些水平或者竖直的连续的空格，用于提示的格子不算空格）不能填重复的数字。当所有格子按要求被填满后，这个数谜就看作被解决了。图1和图2是一个可能的数谜游戏示例。 

　　当然，直接求解数谜问题的话会比较困难。所以现在我们需要解决的是一个更简单的数谜问题。简单数谜的形状是一个(n+1）行乘(m+1)列的矩形。而简单数谜也只有两种要求，就是行要求和列要求，且分别处于第一行和第一列，其他格子则是空格，而左上角是忽略不计的。coolzzz同学爱好简单数谜，他已经给一些简单数谜填好了其中的一些空格。现在，他想寻求你的帮助，来帮他完成这些简单数谜。如图3所示，2和9是coolzzz同学已经填好的空格，图4则是一个基于图3 的一个可能的解答。 

输入

输入包含多组测试数据。第一行包含一个正整数T，表示测试数据数目。每组数据第一行是n(n< 10)和m(m< 10)，表示数谜的形状的大小。接下来一行有n个整数，是相应的行要求；然后一行是m个整数，是相应的列要求。接下来的n行每行有m个小于10的非负整数，0表示该空格还没有被填数字，其他表示coolzzz同学已经填好的数字。输入数据保证未填数字的空格不会超过16个。 
 

输出

对于每组测试数据，输出若干行。如果基于coolzzz已填的结果，该数谜只有一个解，则输出该解；如果不止一个解，则输出一行“Not unique.”；如果没有解，则输出一行“No answer.”。

样例输入

3
3 3
6 6 6
6 6 6
0 0 0
0 3 0
0 0 0
2 3
10 17
5 16 6
2 0 0
0 9 0
2 2
3 5
4 4
0 0
0 0

样例输出

Not unique.
2 7 1
3 9 5
No answer.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int t,m,n,map[15][15];//map数组用来记录初始数谜 
int h[15];//h数组用来记录每行的要求
int l[15];//l数组用来记录每列的要求
int hs[15][15],ls[15][15];//hs数组用来记录每行数字1-9是否用过，ls记录每列是否用过 
int ans[15][15];//用来记录填完之后的数谜 
int num;//num用来记录解的个数 
void dfs(int x,int y)//搜索的坐标 
{
    if(num>1)return;//解多于1个，不需再求 
    if(x==n)//搜索到最后一行
    {
        int flag=1;
        for(int i=0;i<m;i++)//判断每列是否符合要求 
        {
            if(l[i]!=0)
            {
                flag=0;
                break;
            }
        } 
        if(flag)
        {
            num++;//如果满足条件，解的个数+1 
            if(num==1)//如果只有一个解，记录数谜 
            for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
            ans[i][j]=map[i][j];
        }
    }
    else if(y==m)
    {
        if(h[x]==0)
        dfs(x+1,0);//本行所有的列都搜索完并且本行满足要求，搜索下一行 
    }
    else if(map[x][y]!=0)dfs(x,y+1);//如果当前位置有值，搜索本行下一列 
    else
    {
        for(int i=1;i<=9;i++)
        {
            if(!hs[x][i]&&!ls[y][i]&&i<=h[x]&&i<=l[y])//如果本行本列数字i没被放过，并且数字i均满足本行本列的要求
            {
                hs[x][i]=1;//记录本行这个数字用过 
                ls[y][i]=1;//记录本列这个数字用过 
                h[x]-=i;//更新本行要求 
                l[y]-=i;//更新本列要求 
                map[x][y]=i;//更新当前数谜 
                dfs(x,y+1);//搜索下一列 
                hs[x][i]=0;//回溯本行这个数字没用过 
                ls[y][i]=0;//回溯本列这个数字没用过 
                h[x]+=i;//回溯本行要求 
                l[y]+=i;//回溯本列要求
                map[x][y]=0;//回溯数谜 
            } 
        }
    } 
} 
int main()
{
    cin>>t;
    while(t--)
    {
        num=0;
        memset(hs,0,sizeof(hs));//初始化每行的数字1-9都没用过 
        memset(ls,0,sizeof(ls));//初始化每列的数字1-9都没用过 
        memset(map,0,sizeof(map));//初始化输入的数谜 
        memset(ans,0,sizeof(ans));//初始化填完之后的数谜 
        memset(h,0,sizeof(h));//初始化每行的要求 
        memset(l,0,sizeof(l));//初始化每列的要求 
        cin>>n>>m;
        for(int i=0;i<n;i++)
        cin>>h[i];
        for(int i=0;i<m;i++)
        cin>>l[i];
        for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
        {
        cin>>map[i][j];   
        h[i]-=map[i][j];//每行要求除去已经填好的数 
        l[j]-=map[i][j];//每列要求除去已经填好的数 
        hs[i][map[i][j]]=1;//这行设置这个数字无法使用 
        ls[j][map[i][j]]=1;//这列设置这个数字无法使用 
        }
        dfs(0,0);
        if(num==0)cout<<"No answer."<<endl;
        else if(num>1)cout<<"Not unique."<<endl;
        else
        {
           for(int i=0;i<n;i++)
           {
            cout<<ans[i][0];
            for(int j=1;j<m;j++)
             cout<<" "<<ans[i][j];
             cout<<endl;
           }
        } 
    }
    return 0;
}