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

yuanzhen@ubuntu:~/C_script$ cat dfs_eight_queen.cpp

#include

#include

#include

#include

const int M=8;

const int N=8;

using std::cout;

using std::endl;

using std::vector;

vector > vec(M,vector(N,0));

vector > book(M,vector(N,0));

int next[8][2]={{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1},{-1,0},{-1,1}};

std::set > > vset;

void show(vector > avec)

{

vector >::const_iterator cit;

vector::const_iterator it;

for(cit=avec.begin();cit!=avec.end();++cit)

{

vector pvec=*cit;

for(it=pvec.begin();it!=pvec.end();++it)

cout << *it << " ";

cout << endl;

}

cout << endl;

cout << endl;

}

void mark_vec(vector > &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 > avec)

{

int sum=0;

vector >::const_iterator cit;

vector::const_iterator it;

for(cit=avec.begin();cit!=avec.end();++cit)

{

vector pvec=*cit;

for(it=pvec.begin();it!=pvec.end();++it)

sum=sum+(*it);

}

return sum;

}

void dfs(int step, vector > &avec)

{

if(step==8)

{

int sum=get_sum(avec);

std::set > >::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;

}