八皇后——回溯算法（51. N 皇后）

Python ml

已于 2023-06-30 19:36:55 修改

阅读量93

点赞数

分类专栏：算法 LeetCode 文章标签：算法 c++

于 2021-07-02 18:56:55 首次发布

本文链接：https://blog.csdn.net/m0_48711099/article/details/118421679

版权

八皇后问题 C++ 深度优先搜索递归动态规划

关键词由CSDN通过智能技术生成

LeetCode 同时被 2 个专栏收录

180 篇文章 0 订阅

订阅专栏

算法

22 篇文章 0 订阅

订阅专栏

#include<iostream>
using namespace std;

int* queen = new int[8];
int count=0;

bool IsLegal(int currentrow, int currentcolum) {       //验证当前摆放位置是否合法
    for (int precolum = 0; precolum < currentcolum; precolum++) //与之前每一列皇后的摆放位置比较
    {
        int row = queen[precolum];        
        if (row == currentrow) { return false; }   //判断precolum列的摆放位置是否与(currentrow，currentcolum)同一行  
        int rowdiff = abs(row - currentrow);
        int columdiff = abs(precolum - currentcolum);
        if (rowdiff == columdiff) { return false; }   //判断precolum列的摆放位置是否与(currentrow，currentcolum)同一斜线   
    }
    return true;
}
void eight_queen(int currentcolum) {    //在当前列选择皇后摆放的合法行数
    for (int row = 0; row < 8; row++)   //遍历在currentcolum中每行是否合法
    {
        if (IsLegal(row, currentcolum))   //若行数row合法
        {
            queen[currentcolum] = row;      //记录第currentcolum列的合法行数为row
    	    if (currentcolum != 7)			//未到达第8列，递归选择下一列的一个合法位置
    	    {
        	eight_queen(currentcolum+1);
   		    }else{
    		count++;  				 	//到达第8列，棋盘摆放完成，count++ 
	    	}                          //递归完成，返回上一列的for循环，重新选则当前栈内currentcolum的下一个合法行数
        }
	}
}


int main() {
    eight_queen(0);
    printf("%d", count);
    return 0;
}

class Solution {
public:
    vector<vector<string>> res;
    bool isValid(vector<string>&chessboard, int row, int col, int n){
        for(int i=0;i<row;i++){
            if(chessboard[i][col]=='Q')
                return false;
        }
        for (int i = 1; row-i>=0 && col+i<n;++i){
            if(chessboard[row-i][col+i]=='Q')
                return false;
        }
        for(int i = 1; row-i>=0 && col-i>=0;++i){
             if(chessboard[row-i][col-i]=='Q')
                return false;
        }
        return true;
    }
    void dfs(vector<string>&chessboard,int row,int n){
        if(row==n){
            res.push_back(chessboard);
            return;
        }
        for(int col=0;col<n;col++){
            if(isValid(chessboard, row, col, n)){
                chessboard[row][col]='Q';
                dfs(chessboard,row+1,n);
                chessboard[row][col]='.';
            }
        }
    }
    vector<vector<string>> solveNQueens(int n) {
        vector<string> chessboard(n,string(n,'.'));
        dfs(chessboard,0,n);
        return res;
    }
};