leetcode.51:N皇后问题

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

上图为 8 皇后问题的一种解法。

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

每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例:

输入: 4
输出: [
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],

["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]
解释: 4 皇后问题存在两个不同的解法。

提示：

皇后，是国际象棋中的棋子，意味着国王的妻子。皇后只做一件事，那就是“吃子”。当她遇见可以吃的棋子时，就迅速冲上去吃掉棋子。当然，她横、竖、斜都可走一到七步，可进可退。（引用自 百度百科 - 皇后 ）

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-queens
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：
回溯法，基本的思路是遍历每一个节点，如果当前位置能够放置皇后，则放置皇后并且更新整个棋盘的状态，即每个位置是否能够放置皇后，如果到行末都未能在这一行放置皇后，则重新选择上一行的皇后的位置，如此直到最后一行，放置完所有皇后之后，回溯到上一行继续寻找其他的解。
注意：
1. 方向数组一定要仔细，xy要匹配；
2. 可以通过初始化一个字符串的数组来对结果数组进行简化操作；

class Solution {
public:
    static constexpr int dx[] = {-1,1,0,0,-1,-1,1,1};
    static constexpr int dy[] = {0,0,-1,1,-1,1,-1,1};


    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> ret;
        vector<string> sv;
        vector<vector<int>> mark;


        for(int i = 0;i<n;i++)
        {
            mark.push_back(vector<int>());
            string temp = "";
            for(int j = 0;j<n;j++)
            {
                mark[i].push_back(0);
                temp+=".";
            }
            sv.push_back(temp);
        }

        generate(0,n,sv,ret,mark);
        return ret;
    }

    void generate(int row,int n,vector<string> &sv,vector<vector<string>> &ret,vector<vector<int>> &mark)
    {
        if(row==n)
        {
            ret.push_back(sv);
            return;
        }

        for(int i = 0;i<n;i++)
        {
            if(mark[i][row]==0)
            {
                vector<vector<int>> temp = mark;
                putDownTheQueen(i,row,mark,n);
                //printMark(mark);
                sv[row][i]='Q';
                generate(row+1,n,sv,ret,mark);
                mark = temp;
                sv[row][i] = '.';

            }
        }
        return;
    }


    void putDownTheQueen(int x,int y,vector<vector<int>> &mark,int n)
    {
        mark[x][y] = 1;
        for(int i = 1;i<n;i++)
        {
            for(int j = 0;j<8;j++)
            {
                int nx = i*dx[j]+x;
                int ny = i*dy[j]+y;

                if(nx>=0&&nx<n&&ny>=0&&ny<n)
                {
                    mark[nx][ny] = 1;
                }
            }
        }
    }