算法导论2 E 八皇后问题（dfs）

题目：

题目描述

努比亚和苏丹没有子女，所以他要从一些有集成资格的继承者中挑选一个出来继承王位。他希望这个继承者足够聪明，所以他准备了一个西洋棋盘，上面的每个格子中均有一个 1-99 的数字。他又准备了 8 个皇后棋子。

8 皇后的规则就是不能有任何棋子同行或者同列或者同斜线，在满足这个规则的同时，王位继承者还需要让 8 个皇后所在的位置的数字的和是最大的。

输入

输入一个数字 k(k≤20)，代表棋盘的数量。

接下来有 k 个棋盘，每个棋盘有 64 个数字，分成 8 行 8 列输入，具体可见样例，每一个数字均小于 100。

输出

        每一个棋盘对应输出最大的数值， 一共输出 k行

样例输入 Copy

1

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

样例输出 Copy

260

解析：

dfs的经典问题，做法套用模板即可，这里主要想分享一种简单判断旗子是否在同一条斜线上的方法。这是我之前做其他题时学到的，思想比较精妙，因此印象很深。我看到很多人也都在用这种方法，但是如果不画个图真的挺难理解的。

这种方法就是找在同一个斜线上的元素的下标的规律，找到一个表达式，能让在同一条斜线上的元素的下标带入这个表达式的的值相同，如下图：

图中有两种斜线，一种是从右上向左下倾斜的/，另一种是从左上向右下倾斜的\。

我们先探讨/:会发现，在同一条/线上的元素下标i+j都相等，比如途中画的那条斜线:

1+8=2+7=3+6=....=8+1=9;

这样只需用一个bool数组vis[20]记录下，如果vis[9] = true 就代表图中的这条斜线上已经有棋落子了。

我们取最大的情况：8+8=16,因此vis[]数组至少是要开16个的，这也是我代码中vis数组开到了20个的原因。

再来看\斜线，他们的规律是在同一条斜线上的元素的下标相减都相等，即i-j相等。

比如图中画的那条线，1-1=2-2=3-3=4-4=...=8-8=0;

我们取最大和最小的情况,8-1=7，1-8=-7。所以i-j的范围是[-7,7],负数是无法作为数组的下标使用的，所以干脆在表达式上+7，让i+j+7代表一条斜线。

---

代码：

#include<iostream>
#include<bitset>
using namespace std;
int g[9][9]; 
bitset<20> vis1,vis2,vis3;//分别记录竖线，从左下到右上的斜线，从左上到右下的斜线是否有棋子已经占领了 
int ans,temp;//所求的最大值和每种落子方法所获得的收益 
void dfs(int i)//表示传入第几层了 
{
	if(i==9)//说明在第九层，也就是前面八子都已经落下了
	{
		if(ans<temp)
			ans = temp;
	}
	else
	{
		for(int j=1;j<=8;j++)
		{
			if(!vis1[j]&&!vis2[i+j]&&!vis3[i-j+7])//所在位置符合条件 (剪枝)
			{
				temp += g[i][j];
				vis1[j] = true;vis2[i+j] = true;vis3[i-j+7] = true;
				dfs(i+1);
				vis1[j] = false;vis2[i+j] = false;vis3[i-j+7] = false;//回溯 
				temp -= g[i][j];
			}
		}
	} 
}
void solve()
{
	ans = 0;
	temp = 0;
	for(int i=0;i<20;i++)
	{
		vis1 = false;
		vis2 = false;
		vis3 = false;
	}
	for(int i=1;i<=8;i++)
		for(int j=1;j<=8;j++)
			cin>>g[i][j];
	dfs(1);
	cout<<ans<<'\n';
}
int main()
{
	int k;
	cin>>k;
	while(k--)
		solve();
}