【一天一道LeetCode】#51. N-Queens

一天一道LeetCode系列

（一）題目

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..”]

]

（二）解題

不玩國際象棋還真不知道這題的規則。百度了好久才明白。

主要有以下三個：

+ 同一行上隻能有一個皇後

+ 同一列上隻能有一個皇後

+ 兩個皇後之間不能處在同一條對角線上

具體解法看代碼：

/*
首先利用一個數row和數組a[i]確定每一行每一列隻有一個皇後
然後利用一個vector存儲已擺放皇後的位置坐标，每擺一個皇後就與已擺放的皇後進行比較，如果在一條對角線上就不能擺
*/
class Solution {
public:
    vector<vector<string>> ret;
    vector<pair<int, int>> queens;//存放已擺放的皇後的坐标值
    vector<vector<string>> solveNQueens(int n) {
        int *a = new int[n];//確定每一列隻有一個皇後
        memset(a,,n*sizeof(int));
        vector<string> res;
        backtrc(res, a, , n);
        return ret;
    }
    bool isValid(vector<pair<int,int>> queens , int row,int col)//
    {
        if (queens.empty()) return true;
        for (int i = ; i < queens.size();i++)
        {
            if (abs(row- queens[i].first) == abs(col-queens[i].second))
            {
                return false;
            }
        }
        return true;
    }
    void backtrc(vector<string> res, int a[], int row, int n)//row確定每一行隻有一個皇後
    {
        if (row == n)//如果擺放完n行，則退出
        {
            ret.push_back(res);
            return;
        }
        for (int i = ; i < n; i++)
        {
            if (a[i] == &&isValid(queens, row, i))//保證了同一行，同一列，同一對角線隻有一個Q
            {
                a[i] = ;
                string tmp(n, '.');
                tmp[i] = 'Q';
                res.push_back(tmp);
                queens.push_back(pair<int, int>(row, i));
                backtrc(res, a, row + , n);
                //回溯
                a[i] = ;
                queens.pop_back();
                res.pop_back();
            }
        }
    }
};