题目意思:给定8 x 8的一个方格,对于该方格来说每一行上面只能取一个数,然后同行同列同对角线上都是不满足,求出所能得到的最大值
解题思路:8皇后的变形问题,变为求解所有方案数中的最大的和。我们先分序一下,该解空间的结构,由于每一行只能取一,那么我么就可以知道每一行对应的是解空间的一层,那么我么就可以去搜索这个解空间状态树,从第一行开始搜索,如果行数大于8就return,比较最大的值,在搜索函数里面我们用一个for循环枚举每一列,求出满足条件的一列,通过judge函数去判断,只要判断上面和左边和对角线满足即可,因为后面还不知道,这样就能够遍历整个解空间。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <list>
#include <vector>
#include <stack>
#include <algorithm>
using namespace std;
const int MAXN = 9;
int k, ans;
int val[MAXN][MAXN];
int vis[MAXN][MAXN];
//判断函数
int judge(int x, int y) {
int i, j;
//上方向
for (i = x - 1; i >= 1; i--) {
if (vis[i][y])
return 0;
}
//左方向
for (i = y - 1; i >= 1; i--) {
if (vis[x][i])
return 0;
}
//对角线
for (i = 1; i <= 8; i++) {
for (j = 1; j <= 8; j++) {
if (abs(i - x) == abs(j - y)) {//判断对角线的条件
if (vis[i][j])
return 0;
}
}
}
return 1;
}
//深搜回溯求解
void dfs(int i , int max) {
if(i > 8){//如果大于8那么直接更新最大值然后return
if (max > ans)
ans = max;
return;
}
for(int j = 1 ; j <= 8 ; j++){//这一行遍历过去找到满足的点
if (vis[i][j] == 0 && judge(i, j)) {
vis[i][j] = 1;
dfs(i + 1 , max + val[i][j]);//递归到下一行
vis[i][j] = 0;//状态的回溯
}
}
}
//主函数
int main() {
int i, j;
scanf("%d", &k);
while (k--) {
memset(val, 0, sizeof (val));
memset(vis, 0, sizeof (vis));
for (i = 1; i <= 8; i++) {
for (j = 1; j <= 8; j++)
scanf("%d", &val[i][j]);
}
ans = 0;
dfs(1 , 0);
printf("%5d\n", ans);
}
}