N皇后

题目

按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。

给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q''.' 分别代表了皇后和空位。

示例1:

queens

输入:n = 4
输出:[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释:如上图所示,4 皇后问题存在两个不同的解法。

示例 2:

输入:n = 1
输出:[["Q"]]

思路

我没放弃这个题哦~

根据n创建一个二维数组,作为是否可以使用的位置,关键在于确定每个皇后斜列的对应位置和皇后的位置关系,从而改变二维数组的值

class Solution {
public:
    vector<vector<string>> res;
    vector<string> path;
    void update_used(vector<vector<bool>>& unsafe_index, int i, int j, int n)
    {
        int row = i;
        int column = j; 
        //每列的行位置不用改变
        //每行的列位置改变
        for(int k = row + 1; k < n; ++k)  unsafe_index[k][j] = true;
        //左斜边位置改变
        while(true)
        {
            if(row == n-1 || column == 0) break;
            unsafe_index[++row][--column] = true;
        }
        row = i;
        column = j; 
        //右斜边位置改变
        while(true)
        {
            if(row == n-1 || column == n-1) break;
            unsafe_index[++row][++column] = true;
        }
    }
    void solveNQueensAux(int n, vector<vector<bool>> unsafe_index, int iter) 
    {
        if(iter == n) return;
        for(int j = 0; j < n; ++j)
        {
            vector<vector<bool>> unsafe_index_tmp = unsafe_index;
            if(unsafe_index_tmp[iter][j]) continue;
            string queen_index_path = string(n, '.'); 
            queen_index_path[j] = 'Q';
            path.emplace_back(queen_index_path);
            if(path.size() == n) 
            {
                res.emplace_back(path);
            }
            else
            {
                update_used(unsafe_index_tmp, iter, j, n);
                solveNQueensAux(n, unsafe_index_tmp, iter + 1);
            }
            path.pop_back();
        }
    }
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<bool>> unsafe_index(n, vector<bool>(n, false));
        solveNQueensAux(n, unsafe_index, 0);
        return res;
    }
};

一个小时十五分钟win

虽然时间只超过了5.07%的用户

Leave a Comment

Your email address will not be published. Required fields are marked *

Scroll to Top