N-Queens II

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

class Solution {
public:
    bool isValid(vector<string>& svec, int i, int j, int n)
    {
        for(int h = 0; h < j; ++h)
        {
            if(svec[i][h] == 'Q')
            {
                return false;
            }
        }
        for(int k = 0; k < i; ++k)
        {
            if(svec[k][j] == 'Q')
            {
                return false;
            }
        }

        int k = i - 1, h = j - 1;
        while(k >= 0 && h >= 0)
        {
            if(svec[k][h] == 'Q')
            {
                return false;
            }
            k--;
            h--;
        }

        k = i + 1;
        h = j + 1;
        while(k < n && h < n)
        {
            if(svec[k][h] == 'Q')
            {
                return false;
            }
            ++k;
            ++h;
        }

        k = i - 1, h = j + 1;
        while(k >= 0 && h < n)
        {
            if(svec[k][h] == 'Q')
            {
                return false;
            }
            --k;
            h++;
        }

        k = i + 1, h = j - 1;
        while(k < n && h >= 0)
        {
            if(svec[k][h] == 'Q')
            {
                return false;
            }
            ++k;
            --h;  
        }
        return true;
    }
    void help(vector<string>& svec, int dep, int n, int& sum)
    {
        if(dep == n)
        {
            //vsvec.push_back(svec);
            ++sum;
            return;
        }
        for(int i = 0; i < n; ++i)
        {
            if(isValid(svec, i, dep, n))
            {
                svec[i][dep] = 'Q';
                help(svec, dep + 1, n, sum);
                svec[i][dep] = '.';
            }

        }
    }
    int totalNQueens(int n)
    {
        if(n <= 0)
        {
            return 0;
        }
        string s = "";
        for(int i = 0; i < n; ++i)
        {
            s += '.';
        }
        vector<string> svec(n, s);
        int sum = 0;
        help(svec, 0, n, sum);
        return sum;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值