#33 N-Queens

原创 2016年08月29日 09:55:26

题目描述:

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens' placement, where 'Q' and '.' both indicate a queen and an empty space respectively.

Example

There exist two distinct solutions to the 4-queens puzzle:

[
  // Solution 1
  [".Q..",
   "...Q",
   "Q...",
   "..Q."
  ],
  // Solution 2
  ["..Q.",
   "Q...",
   "...Q",
   ".Q.."
  ]
]
题目思路:

这题就用dfs去解,用一个col set去记录被访问过的column,用diag set去记录被访问过的diag。如果这两种都没被访问过,那把Q放在相应的位置,并进行下一行的recursion。

Mycode(AC = 169ms):

class Solution {
public:
    /**
     * Get all distinct N-Queen solutions
     * @param n: The number of queens
     * @return: All distinct solutions
     * For example, A string '...Q' shows a queen on forth position
     */
    vector<vector<string> > solveNQueens(int n) {
        // write your code here
        vector<vector<string>> ans;
        if (n == 0) return ans;
        
        vector<string> sofar;
        unordered_set<int> col, diag;
        solveNQueens(ans, sofar, n, col, diag);
        return ans;
    }
    
    void solveNQueens(vector<vector<string>>& ans,
                      vector<string>& sofar,
                      int n,
                      unordered_set<int>& col,
                      unordered_set<int>& diag)
    {
        if (sofar.size() == n) {
            ans.push_back(sofar);
            return;
        }
        
        string tmp = nDots(n);
        for (int i = 0; i < n; i++) {
            // if there is no Q in same row, col, and diag,
            // then tmp could be a potential ans
            if (col.find(i) == col.end() &&
                !findDiag(diag, sofar.size(), i, n))
            {
                col.insert(i);
                diag.insert(sofar.size() * n + i);
                tmp[i] = 'Q';
                sofar.push_back(tmp);
                
                solveNQueens(ans, sofar, n, col, diag);
                
                sofar.pop_back();
                tmp[i] = '.';
                diag.erase(sofar.size() * n + i);
                col.erase(i);
            }
        }
    }
    
    // write n '.' in a line
    string nDots(int n) {
        string str = "";
        for (int i = 0; i < n; i++) {
            str += ".";
        }
        return str;
    }
    
    // check whether there is Q in diag
    bool findDiag(unordered_set<int>& diag, int i, int j, int n) {
        for (auto it = diag.begin(); it != diag.end(); it++) {
            int row = *it / n;
            int col = *it % n;
            
            if (abs(row - i) == abs(col - j)) return true;
        }
        
        return false;
    }
};


版权声明:三千微尘里,吾宁爱与憎

Lintcode33 N-Queens solution 题解

【题目描述】 The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no...

33.N-Queens-N皇后问题(中等题)

N皇后问题 题目 n皇后问题是将n个皇后放置在n*n的棋盘上,皇后彼此之间不能相互攻击。 给定一个整数n,返回所有不同的n皇后问题的解决方案。 每个解决方案包含一个明确的n皇后放置布局,其...

POJ3239-Solution to the n Queens Puzzle

  • 2011年07月29日 20:30
  • 11KB
  • 下载

Python实现遗传算法求解n-queens问题(1)

Python实现遗传算法求解n-queens问题 本课程将对遗传算法进行简单讲解,通过使用python实现简单的遗传算法求解函数极值以及通过实例了解Pyevolve的使用方法。非常适合熟悉Pyt...

LeetCode---N-Queens

题目大意:将N个皇后摆放在N*N的棋盘上,要求N个皇后彼此间不能相互攻击。 算法思想: 采用一个数组C[n]记录每一行皇后所在位置的列下标,一个数组column[n]记录当前局面以后皇后所在的列,...

leetcode第一刷_N-Queens

八皇后问题应该是回溯法的教学典范。在本科的时候,有一门课叫面向对象,最后的附录有这个问题的源代码,当时根本不懂编程,照抄下来,运行一下出了结果都很开心,哎。 皇后们的限制条件是不能同行同列,也不能同...

【LeetCode】51. N-Queens

51. N-Queens The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that...

LeetCode-N-Queens

作者:disappearedgod 文章出处:http://blog.csdn.net/disappearedgod/article/details/24662201 时间:2014-4-28 ...

LeetCode 50 N-Queens

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens ...

LeetCode52——N-Queens II

LeetCode52——N-Queens II 跟LeetCode51一样,只不过这题是求解解的个数。 这里我套用了上一题的代码,只不过当row==n(递归返回条件)时,不是将一种可行解加入结果集...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:#33 N-Queens
举报原因:
原因补充:

(最多只允许输入30个字)