问题描述
小明最近在教邻居家的小朋友小学奥数,而最近正好讲述到了三阶幻方这个部分,三阶幻方指的是将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
这题我的思路是:先读取输入,把输入幻方中没出现过的数字存入一个数组,然后对这个数组进行全排列,每排列一次,把排列好的数组中的数字依次替代tmp幻方中的0,然后判断这个tmp幻方,如果是可行的三阶幻方,就n++,最后判断n是否为1,如果为1就打印可行幻方,否则输出Too Many。
下面是c++源码:
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<vector>
#define N 9
using namespace std;
bool findNum(vector<int> arr,int num){
for(int i = 0;i < N;i++)
if(arr[i]==num)return true;
return false;
}
//把输入幻方填满
vector<int> func(vector<int> arr,vector<int> nums){
int n = 0;
for(int i = 0;i < N;i++)
if(arr[i]==0)arr[i] = nums[n++];
return arr;
}
//是可行三阶幻方
bool isTrue(vector<int> arr){
int total[8];
total[0] = arr[0]+arr[1]+arr[2];
total[1] = arr[3]+arr[4]+arr[5];
total[2] = arr[6]+arr[7]+arr[8];
total[3] = arr[0]+arr[3]+arr[6];
total[4] = arr[1]+arr[4]+arr[7];
total[5] = arr[2]+arr[5]+arr[8];
total[6] = arr[0]+arr[4]+arr[8];
total[7] = arr[2]+arr[4]+arr[6];
for(int i = 1;i < 8;i++)
if(total[i]!=total[0])return false;
return true;
}
int main(){
vector<int> arr; //存储输入幻方
vector<int> nums; //存储能填的数字
vector<int> ret; //待打印的可行三阶幻方
int i = 0;
int num;
int n = 0; //记录可行的三阶幻方个数
while(i<N){
cin >> num;
if(num>=0&&num<=9)arr.push_back(num),i++;
}
//把没出现的数字放入待排序的nums里
for(i = 1;i < 10;i++)
if(!findNum(arr,i))nums.push_back(i);
do{
vector<int> tmp = func(arr,nums);
if(isTrue(tmp)){ret = tmp;n++;}
}while(next_permutation(nums.begin(),nums.end()));
if(n==1){
for(i = 0;i < N;i++){
cout << ret[i];
if((i+1)%3==0)cout << endl;
else cout << " ";
}
}
else
cout << "Too Many" << endl;
return 0;
}