lintcode: N-Queens

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.

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> > res;    void dfs(vector<int> &solution,int n,int cur){        if(cur==n){            string str="";            for(int i=0;i<n;i++){                str+='.';            }            vector<string> vec;            for(int i=0;i<n;i++){                string tmp=str;                tmp[solution[i]]='Q';                vec.push_back(tmp);            }            res.push_back(vec);            return;        }        for(int i=0;i<n;i++){            solution.push_back(i);            bool ok=true;            for(int j=0;j<cur;j++){                if(i==solution[j]|| (cur+i)==(j+solution[j])|| (cur-i)==(j-solution[j])){                    ok=false;                    break;                }            }            if(ok){                dfs(solution,n,cur+1);                solution.pop_back();            }else{                solution.pop_back();            }        }    }    vector<vector<string> > solveNQueens(int n) {        // write your code here        vector<int> solution;        dfs(solution,n,0);        return res;    }};