1. 题目网址:http://lx.lanqiao.cn/problem.page?gpid=T446
问题描述
小明最近在教邻居家的小朋友小学奥数,而最近正好讲述到了三阶幻方这个部分,三阶幻方指的是将1~9不重复的填入一个3*3的矩阵当中,使得每一行、每一列和每一条对角线的和都是相同的。
三阶幻方又被称作九宫格,在小学奥数里有一句非常有名的口诀:“二四为肩,六八为足,左三右七,戴九履一,五居其中”,通过这样的一句口诀就能够非常完美的构造出一个九宫格来。
4 9 2
3 5 7
8 1 6
有意思的是,所有的三阶幻方,都可以通过这样一个九宫格进行若干镜像和旋转操作之后得到。现在小明准备将一个三阶幻方(不一定是上图中的那个)中的一些数抹掉,交给邻居家的小朋友来进行还原,并且希望她能够判断出究竟是不是只有一个解。
而你呢,也被小明交付了同样的任务,但是不同的是,你需要写一个程序~
输入格式
输入仅包含单组测试数据。
每组测试数据为一个3*3的矩阵,其中为0的部分表示被小明抹去的部分。
对于100%的数据,满足给出的矩阵至少能还原出一组可行的三阶幻方。
输出格式
如果仅能还原出一组可行的三阶幻方,则将其输出,否则输出“Too Many”(不包含引号)。
样例输入
0 7 2
0 5 0
0 3 0
样例输出
6 7 2
1 5 9
8 3 4
数据规模和约定
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 1000ms
2. 思路分析:
对于九宫格的方格我们只能通过尝试的办法来求解,所以需要使用到DFS算法,尝试每一种的可能性,先把九宫格的数字都填完然后再判断每一行、每一列、正对角线、斜对角线的和是否都相等,使用一个变量来记录满足条件的数量,当计数变量等于1的时候那么应该将数组中的元素值记录下来这里我们可以使用一个辅助的数组来进行记录,当循环结束那么所有的结果已经尝试完了,这个时候需要判断这个计数变量是否等于1假如等于1那么直接输出辅助数组里面的元素值,不等于1那么输出"Too Many"
此外需要注意的是还有一个坑就是我们在判断出当前的数字不需要填的时候那么不需要把判断当前已经有数字的代码写进循环里面,因为这样会导致在for循环中对于相同的数据重复递归了很多次,比如像下面的7之前已经有数字了那么假如写进循环里面由于写进了循环里面那么会重复递归了相同的答案很多次这样造成了本来是一个答案的会导致相同的答案重复多次导致错误
if(arr[cur] != 0){
dfs(cur + 1);
}
0 7 2
0 5 0
0 3 0
for(int i = 0; i < 9; i++){
if(arr[cur] == 0 && judge(i + 1)){
arr[cur] = i + 1;
dfs(cur + 1);
//回溯
arr[cur] = 0;
}
}
3. 代码如下:
import java.util.Scanner;
public class Main {
static int arr[] = new int[9];
static int ans[] = new int[9];
static int flag = 0;
static int count = 0;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
for(int i = 0; i < 9; i++){
arr[i] = sc.nextInt();
}
dfs(0);
if(flag == 1){
for(int i = 0; i < 9; i++){
if(i % 3 == 0 && i != 0)
System.out.print("\n");
System.out.print(ans[i] + " ");
}
}else{
System.out.println("Too Many");
}
sc.close();
}
private static void dfs(int cur) {
if(cur == 9){
if(judgeSum()){
flag++;
if(flag == 1){
for(int i = 0; i < 9; i++){
ans[i] = arr[i];
}
}
}
return;
}
if(arr[cur] != 0){
dfs(cur + 1);
}else{
for(int i = 0; i < 9; i++){
if(arr[cur] == 0 && judge(i + 1)){
arr[cur] = i + 1;
dfs(cur + 1);
//回溯
arr[cur] = 0;
}
}
}
}
private static boolean judge(int index) {
for(int i = 0; i < 9; i++){
if(arr[i] == index)return false;
}
return true;
}
private static boolean judgeSum() {
int sum1 = arr[0] + arr[1] + arr[2];
int sum2 = arr[3] + arr[4] + arr[5];
int sum3 = arr[6] + arr[7] + arr[8];
int sum4 = arr[0] + arr[3] + arr[6];
int sum5 = arr[1] + arr[4] + arr[7];
int sum6 = arr[2] + arr[5] + arr[8];
int sum7 = arr[0] + arr[4] + arr[8];
int sum8 = arr[2] + arr[4] + arr[6];
return (sum1 == sum2 && sum2 == sum3 && sum3 == sum4 && sum4 == sum5 && sum5 == sum6 && sum6 == sum7
&& sum7 == sum8);
}
}