c++ 深度优先算法解决八皇后问题

2019独角兽企业重金招聘Python工程师标准>>>

yuanzhen@ubuntu:~/C_script$ cat dfs_eight_queen.cpp
#include <iostream>
#include <vector>
#include <cstdlib>
#include <set>

const int M=8;
const int N=8;

using std::cout;
using std::endl;
using std::vector;

vector<vector<int> > vec(M,vector<int>(N,0));
vector<vector<int> > book(M,vector<int>(N,0));
int next[8][2]={{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1},{-1,0},{-1,1}};
std::set<vector<vector<int> > > vset;
void show(vector<vector<int> > avec)
{
    vector<vector<int> >::const_iterator cit;
    vector<int>::const_iterator it;

    for(cit=avec.begin();cit!=avec.end();++cit)
    {
        vector<int> pvec=*cit;
        for(it=pvec.begin();it!=pvec.end();++it)
            cout << *it << " ";
        cout << endl;
    }
    cout << endl;
    cout << endl;
}

void mark_vec(vector<vector<int> > &avec, int x, int y, int lable)
{
    for(int i=0;i<8;++i)
    {
        int tx=x;
        int ty=y;

        int dx=next[i][0];
        int dy=next[i][1];

        while(tx>=0 && tx<=N-1 && ty>=0 && ty<=N-1)
        {
            avec[tx][ty]=avec[tx][ty]+lable;
            tx=tx+dx;
            ty=ty+dy;
        }
    }
}

int get_sum(vector<vector<int> > avec)
{
    int sum=0;

    vector<vector<int> >::const_iterator cit;
    vector<int>::const_iterator it;

    for(cit=avec.begin();cit!=avec.end();++cit)
    {
        vector<int> pvec=*cit;
        for(it=pvec.begin();it!=pvec.end();++it)
          sum=sum+(*it);
    }
    return sum;
}

void dfs(int step, vector<vector<int> > &avec)
{
    if(step==8)
    {
        int sum=get_sum(avec);
        std::set<vector<vector<int> > >::iterator sit;
        sit=vset.find(avec);
        if(sum==8 && sit==vset.end())
          vset.insert(avec);
        return ;
    }

    for(int i=0;i<8;++i)
    {
    //int i=step;
        for(int j=0;j<8;++j)
        {
            if(book[i][j]==0)
            {
                //show(book);
                mark_vec(book,i,j,1);
                vec[i][j]=1;
                dfs(step+1, avec);
                mark_vec(book,i,j,-1);
                vec[i][j]=0;
            }
        }
    }
    return ;
}

int main(void)
{
    show(vec);
    dfs(0, vec);

    cout << vset.size() << endl;
}
 

转载于:https://my.oschina.net/lCQ3FC3/blog/824729