[LeetCode 51]N皇后

[LeetCode 51]C++实现N皇后

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

给定一个整数 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：

设立方向数组：(x↓ y→)
(x,y)表示皇后所在位置，
方向数组按照8个方向延伸到边界，表示攻击范围
(x-1,y-1) (x-1,y) (x-1,y+1)
 (x,y-1)   (x,y)   (x,y+1)
(x+1,y-1) (x+1,y) (x+1,y+1)  

例：mark(3,3)
1 0 0 1 0 0 1 0
0 1 0 1 0 1 0 0
0 0 1 1 1 0 0 0
1 1 1 1 1 1 1 1 
0 0 1 1 1 0 0 0
0 1 0 1 0 1 0 0
1 0 0 1 0 0 1 0
0 0 0 1 0 0 0 1

步骤2：
以4皇后为例
每一行必须放一个皇后，我们只是选择放在哪一列
依次递归直到皇后的个数为N

实现代码

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> result;//最终结果
        vector<string> location;//存储摆放结果
        vector<vector<int>> mark;//标记是否可以放皇后

        for(int i=0;i<n;++i){
            mark.push_back(vector<int>());//摆放 NxN棋盘
            //mark初始值都为0
            for(int j=0;j<n;++j){
                mark[i].push_back(0);
            }
            //location初始化
            location.push_back("");
            location[i].append(n,'.');//result[i]初始化都为.
        }
        generate(0,n,location,result,mark);
        return result;        
    }
    void generate(int k,int n,vector<string>& location,vector<vector<string>>& result,vector<vector<int>>& mark){
        //k代表行数，n代表皇后个数
        if(k==n){
            result.push_back(location);
            return;
        }
        for(int i=0;i<n;++i){
            if(mark[k][i]==0){
                vector<vector<int>> temp_mark=mark;
                location[k][i]='Q';
                put_down_queen(k,i,mark);//放置皇后
                generate(k+1,n,location,result,mark);
                mark=temp_mark;
                location[k][i]='.';
            }
        }

    }
    void put_down_queen(int x,int y,vector<vector<int>>& mark){
        //方向数组
        static const int dx[]={-1,1,0,0,-1,-1,1,1};
        static const int dy[]={0,0,-1,1,-1,1,-1,1};
        mark[x][y]=1;//标记皇后
        //延伸的长度为N-1
        for(int i=1;i<mark.size();++i){
            for(int j=0;j<8;++j){
                //延伸方向数组
                int new_x=x+i*dx[j];
                int new_y=y+i*dy[j];
                if(new_x>=0&&new_y>=0&&new_x<mark.size()&&new_y<mark.size()){
                    mark[new_x][new_y]=1;
                }
            }
        }
    }

};

[LeetCode 52]N皇后II
方法一致，count计数

class Solution {
public:
    int totalNQueens(int n) {
        vector<vector<int>>mark;
        for(int i=0;i<n;++i){
            mark.push_back(vector<int>());
            for(int j=0;j<n;++j){
                mark[i].push_back(0);
            }
        }
        int count=0;
        recursive(count,0,n,mark);
        return count;
    }
    void recursive(int& count,int k,int n,vector<vector<int>>& mark){
        if(k==n){
            ++count;
            return;
        }
        for(int i=0;i<n;++i){
            if(mark[k][i]==0){
                vector<vector<int>>temp_mark=mark;
                put_down_queen(k,i,mark);
                recursive(count,k+1,n,mark);
                mark=temp_mark;
            }
        }
    }
    void put_down_queen(int x,int y,vector<vector<int>>& mark){
        static const int dx[]={-1,1,0,0,-1,-1,1,1};
        static const int dy[]={0,0,-1,1,-1,1,-1,1};
        mark[x][y]=1;
        for(int i=1;i<mark.size();++i){
            for(int j=0;j<8;++j){
                int new_x=x+i*dx[j];
                int new_y=y+i*dy[j];
                if(new_x>=0&&new_x<mark.size()&&new_y>=0&&new_y<mark.size()){
                    mark[new_x][new_y]=1;
                }
            }
        }
    }
};