也是好久没更新算法题目了,最近也在忙于操作系统实验以及看srp的论文的过程。恰好今天上了算法课,老师也讲了新的内容——回溯法。听说也是考试有概率考到的题目,花了点时间复现一下,顺便讲一下自己的思路。
首先可以给出一个具体的题目:八皇后问题(竟然还是北京大学考研机试题),内容和输出格式具体如下:
可能题目说起来没有图片直观,我在网上找了一个四皇后的图片,可以看到其中一种正确的放置方式应该是这样的:
Ok,那了解到大概需要什么样的正确答案后,我们讲讲思路。大概也是老师课上的复盘。首先暴力解法当然是不可行的,因为总共有8的8次方种方式,其中很多方式一看上去是错的,而且重复的计算很多。所以必须要使用一定的算法。那么为什么会想到回溯法呢??(废话,不然老师为什么会讲到这个~)在我自己解了一遍题目过后发现,其实回溯法跟动态规划有点类似的是:它们都减少了很多重复的计算!当然这是我个人小白的观点。
那么这个回溯法到底怎么减少了重复计算呢?举个例子假如第一行的皇后选了第1列,而第2行的皇后也选了第1列,实际上这种情况下,后面的皇后根本不用继续排了!但是暴力法的话就发现不了这个问题。也就是说,如果我们能写一个判断函数来判断当前情况是否需要继续下去,如果是则继续,如果否则直接去掉后面所有的情况,那么问题就可以简化很多了。这个函数在下面的代码中体现为Condition函数。
而对每个结点来说,也是顺序遍历(例如第一个皇后选第1列,第2列一直到第8列,然后第一个皇后在选定第某一列后,第二个皇后又开始选从第1列到第8列,如此套娃循环下去)。虽然看起来这道题需要用到指针,树等结点(至少我第一次看到的时候是这样)。但实际上我们只要用一个数组记录每个皇后(对应每一行)在哪一列就行了。只要遍历过程中一直去掉错误的结点,自然最后留下来的就是正确的答案,所以我们在最后一个节点通过检验的情况下,记录所有的值即可。
最后的输出结果说是要按从小到大的顺序写出编号对应的数,其实我们在上述过程中已经是按编号遍历的,最后只需要将所有节点的值合起来就是排好序的数组了。由于题目已经给出皇后数量为8,最终结果数为92所以我们直接开个全局变量数组就行了(多一个是方便从1开始编号)。至于n皇后的情况大家可以类比着写一下,应该是差不多的。
下面是根据老师给出的伪码写出的C++完整代码(有不懂的可以评论区交流~):
#include<iostream> #include<cmath> #include<string> using namespace std; const int N = 8;//皇后的个数 int answer[93];//解的个数 int col[9];//皇后所在的列数,此时为全局变量,自动初始化为0 int counter = 1; bool Condition(int queen) { int i = 1; while (i < queen) { if (col[i] == col[queen] || abs(col[i] - col[queen]) == abs(i - queen)) { return false; } i++; } return true; }//用来判断皇后位置是否符合约束条件,符合返回真,否则返回假 void Trace() { int k = 1;//用来记录回溯的结点编号 while (k > 0) { col[k] = col[k] + 1;//表示从第一列开始遍历,往后逐渐加一 while (col[k] <= N && !Condition(k)) { col[k] = col[k]+1;//如果未遍历完,而且当前列不符合约束条件,则直接换列 }//退出循环则说明当前列符合条件或者直接全部列不合适!究竟是哪种情况用if判断一下 if (col[k] <= N) { if(k==N){ string temp = ""; for (int i = 1; i <= N; i++) { temp += to_string(col[i]); } answer[counter++] = atoi(temp.c_str()); } else { k = k + 1; col[k] = 0;} }//说明这是当前列符合条件的情况 else { k = k - 1; } } } int main() { int n, no; cin >> n ; Trace(); for (int i = 0; i < n; i++) { cin >> no; cout << answer[no] << endl; } }