东华OJ 基础题131 8皇后·改——难度易

问题描述 :

规则同8皇后问题，但是棋盘上每格都有一个数字，要求八皇后所在格子数字之和最大。

输入说明 :

一个8*8的棋盘。

数据规模和约定

棋盘上的数字范围0~99

输出说明 :

所能得到的最大数字和

输入范例 :

1 2 3 4 5 6 7 8
9 10 11 12 13 14 15 16
17 18 19 20 21 22 23 24
25 26 27 28 29 30 31 32
33 34 35 36 37 38 39 40
41 42 43 44 45 46 47 48
48 50 51 52 53 54 55 56
57 58 59 60 61 62 63 64

输出范例 :

260

完整C++代码

都在注释里😆

#include <bits/stdc++.h>

using namespace std;

int n = 8, max_sum = 0;
int point[8][8];
// res[1]=3，表示第2行第4列，这样可以将二维压缩成一维 
int res[8] = {0}, col[8] = {0}, l_up[30] = {0}, l_down[30] = {0};

void dfs(int num) {
	if (num == n) {
		// 得到一种八皇后的放置方案 
		int sum = 0;
		// 看上面res数组的解释，算出目前方案的总和 
		for (int i = 0; i < n; i++) {
			sum += point[i][res[i]];
		}
		// 取最大值 
		max_sum = max(max_sum, sum);
		return;
	}
	for (int j = 0; j < n; j++) {
		// 第j列、主对角线和副对角线为0才可放置 
		if (col[j] || l_up[j - num + n] || l_down[j + num]) continue;
		// 可放置 
		res[num] = j;
		col[j] = 1;
		l_up[j - num + n] = 1;
		l_down[j + num] = 1;
		// 放置 
		dfs(num + 1);
		// 回溯
		col[j] = 0;
		l_up[j - num + n] = 0;
		l_down[j + num] = 0;
	}
}

int main()
{
	// 棋盘数字
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> point[i][j];
		}
	}
	dfs(0);
	cout << max_sum << endl;
	return 0;
}

😋欢迎大伙私信或者评论区交流讨论😋