东华OJ 进阶题16 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;

typedef vector<int> vec;
typedef vector<vec> mat;

int n = 8, max_sum = 0;
int res[8] = {0}, col[8] = {0}, l_up[30] = {0}, l_down[30] = {0};
mat points(8, vec(8));

void dfs(int step) {
	if (step == n) {
		int sum = 0;
		for (int i = 0; i < n; i++) {
			sum += points[i][res[i]];
		}
		max_sum = max(max_sum, sum);
		return;
	}
	for (int i = 0; i < n; i++) {
		if (col[i] || l_up[i - step + n] || l_down[i + step]) continue;
		res[step] = i;
		col[i] = 1;
		l_up[i - step + n] = 1;
		l_down[i + step] = 1;
		dfs(step + 1);
		col[i] = 0;
		l_up[i - step + n] = 0;
		l_down[i + step] = 0;
	}
}

int main()
{
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> points[i][j];
		}
	}
	dfs(0);
	cout << max_sum << endl;
	return 0;
}

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