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