LeetCode - 52. N-Queens II

52. N-Queens II 

Problem's Link

 ----------------------------------------------------------------------------

Mean: 

略.

analyse:

略

Time complexity: O(N)

 

view code

/**
* -----------------------------------------------------------------
* Copyright (c) 2016 crazyacking.All rights reserved.
* -----------------------------------------------------------------
*       Author: crazyacking
*       Date  : 2016-03-21-11.01
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long( LL);
typedef unsigned long long( ULL);
const double eps( 1e-8);

class Solution
{
public :
    int totalNQueens( int n)
    {
        ans = 0;
        mat =* new vector < string >(n , string(n , '.'));
        dfs( mat , 0 ,n);
        return ans;
    }
    void dfs( vector < string > & mat , int row , int n)
    {
        if( row ==n)
            { ans ++; return ;}
        for( int col = 0; col <n; ++ col)
        {
            if( check( row , col , mat ,n))
            {
                mat [ row ][ col ] = 'Q';
                dfs( mat , row + 1 ,n);
                mat [ row ][ col ] = '.';
            }
        }
    }

    bool check( int row , int col , vector < string > & mat , int n)
    {
        for( int i = 0; i < row; ++ i)
            if( mat [ i ][ col ] == 'Q')
                return 0;
        for( int x = row - 1 , y = col - 1; x >= 0 && y >= 0; -- x , -- y)
            if( mat [ x ][ y ] == 'Q')
                return 0;
        for( int x = row - 1 , y = col + 1; x >= 0 && y <n; -- x , ++ y)
            if( mat [ x ][ y ] == 'Q')
                return 0;
        return 1;
    }

private :
    int ans;
    vector < string > mat;
};

int main()
{
    int n;
    while( cin >>n)
    {
        Solution solution;
        int ans = solution . totalNQueens(n);
        cout << ans << endl;
        cout << "------------------------------" << endl;
    }
    return 0;
}
/*

*/