蓝桥杯 九宫幻方

蓝桥杯 九宫幻方 C++

思路

1.将所有满足条件的3阶幻方列举出来
使用全排列函数next_permutation来实现,函数使用链接
https://blog.youkuaiyun.com/Dear_Jia/article/details/79593973

2.遍历所有3阶幻方矩阵，统计与输入矩阵可以匹配的个数

3.代码

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <bits/stdc++.h>
using namespace std;
int main() {
	int n=3;
	int a[n+1][n+1];
	for(int i=1; i<=3; i++)
		for(int j=1; j<=3; j++)
			cin>>a[i][j];
	int num[10]= {0,1,2,3,4,5,6,7,8,9};
	int squre[20][10];//第二维用于保存幻方矩阵 
	int count=0;//三阶幻方个数
	//将所有的幻方矩阵列举并保存 
	do {
		if(num[1]+num[2]+num[3]==15&&num[4]+num[5]+num[6]==15&&num[7]+num[8]+num[9]==15)
			if(num[1]+num[4]+num[7]==15&&num[2]+num[5]+num[8]==15&&num[3]+num[6]+num[9]==15)
				if(num[1]+num[5]+num[9]==15&&num[3]+num[5]+num[7]==15) {
					for(int i=1; i<=9; i++)
						squre[count][i]=num[i];
					count++;
				}
	} while(next_permutation(num+1,num+10));//全排列函数 
	int sum=0,flag;
	for(int i=0; i<count; i++) {
		int  temp=0;
		for(int j=1; j<=3; j++) {
			for(int k=1; k<=3; k++) {
				int t=3*(j-1)+k;
				if((a[j][k]==squre[i][t])||a[j][k]==0)
					temp++;
			}
		}
		if(temp==9) {//输入的矩阵如果和某一个幻方矩阵一样，sum加一 
			sum++;
			flag=i;//保存幻方矩阵的序号，用于输出 
		}

	}
	if(sum==1) {//如果只有一个就输出 
		for(int i=1; i<=9; i++) {
			cout<<squre[flag][i]<<" ";
			if(i%3==0)
				cout<<endl;
		}
	}
	if(sum>1)
		cout<<"Too Many"<<endl;
	return 0;
}