[leetcode][52. N皇后 II]

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

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

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

示例:

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

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]

 

方法叫回溯？有一步回退代码时忘了改状态值了，调试了好久，坑。。。

class Solution {
public:
    int totalNQueens(int n) {
        vector<int> data(n,0);
        vector<int> linex(n,0);
        vector<int> liney(n,0);
        vector<int> linem(2*n,0);
        vector<int> linen(2*n,0);
        int result=0;
        int pos = 0;
        while(pos>=0 && pos<n)
        {
            
            if(data[pos] == 0) ++data[pos];  
            while(data[pos]<=n && (linex[data[pos]-1]>0 || liney[pos]>0 || linem[data[pos]-1+pos]>0 || linen[pos+n-data[pos]]>0))
            {
                ++data[pos];
            }
            
            //if over board,jump back
            if(data[pos] > n)
            {
                data[pos] = 0;
                --pos;
                
                //judge if out
                if (pos<0) return result;
                
                //jump back
                --linex[data[pos]-1];
                --liney[pos];
                --linem[data[pos]-1+pos];
                --linen[pos+n-data[pos]];
                ++data[pos];
                continue;
            }
            
            
            //update state
            ++linex[data[pos]-1];
            ++liney[pos];
            ++linem[data[pos]-1+pos];
            ++linen[pos+n-data[pos]];

            //judge correct answer
            if (pos==n-1)
            {
                //count result
                ++result;
                
                //jump back
                --linex[data[pos]-1];
                --liney[pos];
                --linem[data[pos]-1+pos];
                --linen[pos+n-data[pos]];
                ++data[pos];
                continue;
            }
            
            //normal condition
            ++pos;
            
        }
        return result;
    }
};