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

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);

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