蓝桥杯 PREV-42 九宫幻方（dfs）

题目链接：

PREV-42 九宫幻方

思路：

直接dfs搜索所有可能，记录能形成幻方的矩阵即可；

代码：

#include<bits/stdc++.h>

using namespace std;

int cnt, vst[10];
vector<vector<int> > mat(3, vector<int>(3)), ans(3, vector<int>(3));

bool check() {
    
    
	for(int i = 0; i < 3; i++) {
    
    
		if(mat[i][0] + mat[i][1] + mat[i][2] != 15) return false;
		if(mat[0][i] + mat[1][i] + mat[2][i] != 15) return false;
	}
	if(mat[0][0] + mat[1][1] + mat[2][2] != 15) return false;
	if(mat[0][2] + mat[1][1] + mat[2][0] != 15) return false;
	return true;
}
void dfs(int pos) {
    
    
	if(pos == 9) {
    
    
		if(check()) ++cnt, ans = mat;
		if(cnt > 1) cout << "Too Many", exit(0);
		return;
	}
	int x = pos / 3, y = pos % 3;
	if(mat[x][y]) dfs(pos + 1);
	else for(int i = 1; i <= 9; i++) if(!vst[i]) {
    
    
		mat[x][y] = i, vst[i] = true;
		dfs(pos + 1);
		mat[x][y] = 0, vst[i] = false;	
	}
}

#define TRA for(int i = 0; i < 3; i++) for(int j = 0; j < 3; j++)
int main() {
    
    
#ifdef MyTest
	freopen("Sakura.txt", "r", stdin);
#endif
	TRA cin >> mat[i][j], vst[mat[i][j]] = true;
	mat[1][1] = 5, vst[5] = true;
	dfs(0);
	TRA cout << ans[i][j] << (j == 2 ? '\n' : ' ');
	return 0;
}