51. N-Queens

2017-10-11  本文已影响0人  Al73r

题目

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.


Given an integer n, return all distinct solutions to the n-queens puzzle.
Each solution contains a distinct board configuration of the n-queens' placement, where 'Q'
and '.'
both indicate a queen and an empty space respectively.
For example,There exist two distinct solutions to the 4-queens puzzle:
[
 [".Q..",  // Solution 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // Solution 2
  "Q...",
  "...Q",
  ".Q.."]
]

分析

n皇后问题,比较经典。没记错的话应该使用回溯法。一层层往下搜索,跳过不能放置的位置,如果个数够了就输出,如果搜完全图都不够就返回。另外对于斜线的表示,可以利用主斜线上元素的坐标之差固定,副斜线上元素的坐标之和固定这个性质来判断。

实现一

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> ans;
        vector<string> board(n, string(n, '.'));
        vector<bool> row(n, false), col(n, false);
        vector<bool> md(2*n-1, false), cd(2*n-1, false);
        solve(0, 0, n, ans, board, row, col, md, cd);
        return ans;
    }
private:
    void solve(int x, int y, int nleft, vector<vector<string>> &ans,
               vector<string> &board, vector<bool> &row,
               vector<bool> &col, vector<bool> &md, vector<bool> &cd){
        if(nleft==0){
            ans.push_back(board);
            return;
        }
        if(y>board[0].size()-1){
            if(x<board.size()-1){
                x++;
                y = 0;
            }
            else{
                return;
            }
        }
        for(int i=x; i<board.size(); i++){
            if(row[i]) continue;
            for(int j=((i==x)?y:0); j<board[0].size(); j++){
                if(col[j] || md[i-j+board.size()-1] || cd[i+j]) continue;
                board[i][j] = 'Q';
                row[i] = true;
                col[j] = true;
                md[i-j+board.size()-1] = true;
                cd[i+j] = true;
                
                solve(i, j+1, nleft-1, ans, board, row, col, md, cd);
                board[i][j] = '.';
                row[i] = false;
                col[j] = false;
                md[i-j+board.size()-1] = false;
                cd[i+j] = false;
                
            }
        }
    }
};

思考一

通过是通过了,但是相比别人3ms的解答,我这132ms的方法简直腊鸡到我不忍直视。看了别人的代码受到了启发:皇后的性质决定了棋盘的每一行有且只有一个皇后,所以直接枚举每一行就行了。这一简单的改动不但使得代码更加简单了,而且让复杂度由原来的O(n(n2))降低到了O(n^n),直接让程序的耗时到达了3ms。

实现二


思考二

由于我的代码经过了多次修改,所以可能不太美观,见谅=_=。

上一篇下一篇

猜你喜欢

热点阅读