LintCode-33: N-Queens (DFS排列经典题!)

原创已于 2023-07-09 14:54:02 修改 · 336 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LintCode

于 2018-07-21 14:58:30 首次发布

algorithm-design 专栏收录该内容

817 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）解决N皇后问题的方法，并详细解释了如何通过一个isValid()函数来检查列冲突及对角线冲突，确保放置的皇后不会互相攻击。此外还提供了一种优化方案，利用三个数组跟踪列、对角线和反对角线的状态，以降低检查冲突的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

又是一道DFS排列经典题。这题用visited[]还不够，要专门用一个isValid()函数来确保没有列冲突和两个对角线冲突。

注意：
1） string的初始化：
string dotStr(n, ‘.’);
vector solString(n, dotStr); //good string initialization code!

代码如下：

class Solution {
public:
    /*
     * @param n: The number of queens
     * @return: All distinct solutions
     */
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> results;
        vector<int> sol;
        if (n == 0) {
            results.push_back(vector<string>());
            return results;
        }
        
        helper(n, 0, sol, results);
        return results;
    }
    
private:
    bool isValid(vector<int> &sol, int col) {
        int row = sol.size();
        for (int rowId = 0; rowId < sol.size(); ++rowId) {
            if (sol[rowId] == col) 
                return false;
            if (sol[rowId] + rowId == col + row)
                return false;
            if (sol[rowId] - rowId == col - row)
                return false;
        }
        return true;
    }
    
    void helper(int n, int index, vector<int>&sol, vector<vector<string>> &results) {
        if (index == n) {
            string dotStr(n, '.');
            vector<string> solString(n, dotStr);   //good string initialization code!
            for (auto i : sol) {
                solString[i][sol[i]] = 'Q';
            }
            results.push_back(solString);
            return;
        }
        
        for (int i = 0; i < n; ++i) {
            
            if (isValid(sol, i)) {
                sol.push_back(i);
                helper(n, index + 1, sol, results);
                sol.pop_back();
            }
            
        }
    }
    
};

写成下面这样也可以。这样for loop里面用sol[index] = i; 就可以了，不用push_back又pop_back。

class Solution {
public:
    /*
     * @param n: The number of queens
     * @return: All distinct solutions
     */
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> results;
        vector<int> sol(n, 0);
        if (n == 0) {
            results.push_back(vector<string>());
            return results;
        }

        helper(n, 0, sol, results);
        return results;
    }

private:
    bool isValid(vector<int> &sol, int col, int index) {
        int row = index; //sol.size();
        for (int rowId = 0; rowId < index; ++rowId) {
            if (sol[rowId] == col) 
                return false;
            if (sol[rowId] + rowId == col + row)
                return false;
            if (sol[rowId] - rowId == col - row)
                return false;
        }
        return true;
    }

    void helper(int n, int index, vector<int>&sol, vector<vector<string>> &results) {
        if (index == n) {
            string dotStr(n, '.');
            vector<string> solString(n, dotStr);   //good string initialization code!
            for (auto i : sol) {
                solString[i][sol[i]] = 'Q';
            }
            results.push_back(solString);
            return;
        }

        for (int i = 0; i < n; ++i) {

            if (isValid(sol, i, index)) {
                sol[index] = i;
                helper(n, index + 1, sol, results);
            }

        }
    }
};

另外，这题还可以优化。主要是isValid()的复杂度可以从O(n)降到O(1)。我们可以设3个长度为n…2n的数组，分别表示列，对角线和反对角线是否有冲突。反对角线可以用i+j，即如果i行j列有Queen，则a[i+j]=1。对角线则要用i-j或j-i。但这里有个问题因为i-j或j-i可能为负。一个技巧就是用i-j+n，这样就没有负数了。