leetcode--52N皇后二

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。
给定一个整数 n，返回 n 皇后不同的解决方案的数量。

示例:
输入: 4
输出: 2
解释: 4 皇后问题存在如下两个不同的解法。
[
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],

["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-queens-ii

方法：
这道题相比于51题要简单很多，只需要得到结果，而不需要知道解的形式，所以我们参考51题，只需要去掉不需要的参数即可。终止条件改为当索引可以到n时，证明0，n-1都可以放入皇后，那就是一组解了。

class Solution {
public:
    int res = 0;
    vector<bool> col, dia1, dia2;
    int totalNQueens(int n) {
        col = vector<bool> (n, false);
        dia1 = vector<bool> (2*n-1, false);
        dia2 = vector<bool> (2*n-1, false);
        putqueen(n,0);
        return res;

    }

    void putqueen(int n, int index){
        if(index == n){
            res++;
            return;
        }

        for(int i = 0; i < n; i++){
            if(!col[i] && !dia1[index + i] && !dia2[index-i+n-1]){
                col[i] = true;
                dia1[index+i] = true;
                dia2[index-i+n-1] = true;
                putqueen(n,index+1);
                col[i] = false;
                dia1[index+i] = false;
                dia2[index-i+n-1] =false;
            }
        }
        return;
    }

};