关于回溯法的递归与非递归—–N皇后问题
蓝桥杯 基础练习 2n皇后问题
给定一个n*n的棋盘,棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后,使任意的两个黑皇后都不在同一行、同一列或同一条对角线上,任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法?n小于等于8。
输入格式
输入的第一行为一个整数n,表示棋盘的大小。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,如果一个整数为0,表示对应的位置不可以放皇后。
输出格式
输出一个整数,表示总共有多少种放法。
样例输入
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
2
样例输入
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
0
最初思路(运行超时)
定义一个二维数组当棋盘,挨个遍历,每个位置分为放黑皇后或放白皇后或不放三个选择;用回溯法的递归实现。
后来看了别人的代码,自己的方法简直太笨了,分析了一下有以下缺点:
- 黑白一起考虑:这样每个位置就多了一个选择,在递归的过程相当于时间复杂度多乘了个n;如果先考虑黑在考虑白,相当于时间复杂度乘以2。
- 二维数组:遍历麻烦,而且耗时。因为一行只能放一个皇后(先只考虑一个颜色的皇后)所以这一行只要有皇后了,就直接到下一行去找。
- 递归:通常来说回溯法分为递归和非递归,递归的效率低一些。要练习使用非递归的方法。
改进方法
- 黑白分开考虑,先看黑皇后,再看白。
- 用一位数组代替,q[i]代表第i行上的皇后在第几列。
- 用非递归。
#include <stdio.h>
#include <iostream>
using namespace std;
int blq[8];//每行的黑皇后在第几列
int whq[8];//每行的白皇后在第几列
int n;
int can[8][