Eight Queen

八皇后问题

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <cmath>

using namespace std;
int p[8][8];                // 棋盘数组
int cnt;                    //记录解的个数

//判断(r,c)位置能否放置棋子,能,返回true,不能返回false
//因为是从第一行开始往下放棋子，所以只需要判断r行之前的行中有没有棋子使得(r,c)不可落子即可
bool can(int r, int c){
for (int i=0; i<r; i++){        //判断同一列中是否有棋子，有则返回false
if (p[i][c] == 1)
return false;
}

int rr;                         //临时纵坐标(行数)
int rc;                         //临时横坐标(列数)

for (int i=1; i<=r; i++){       //判断(r,c)左上方是否有棋子使(r,c)处不可落子
rr = r-i;
rc = c-i;

if (rr>=0 && rc>=0){
if (p[rr][rc] == 1)
return false;
}
}

for (int i=1; i<=r; i++){       //判断(r,c)右上方是否有棋子使(r,c)处不可落子
rr = r-i;
rc = c+i;

if (rr>=0 && rc<=7){
if (p[rr][rc] == 1)
return false;
}
}

return true;
}

//显示各个结果
void show(){
for (int i=0; i<8; i++){
for (int j=0; j<8; j++){
printf("%d ", p[i][j]);
}
printf("\n");
}
}

//dfs遍历，参数r为落子的纵坐标(行)
void dfs(int r){

if (r == 7){                        //若这是最后一行，则在此行中找到可以落最后一个棋子的位置
for (int j=0; j<8; j++){
if (can(r, j)){
p[r][j] = 1;            //将该位置设为有棋子
cnt++;                  //解的个数加一
printf("%d\n", cnt);    //输出解的编号
show();                 //显示改解
p[r][j] = 0;            //回溯，将该位置设为没有棋子
}
}
}else{
for (int j=0; j<8; j++){        //若不是最后一行，则在该行中找到可已落子的位置，再从下一行开始执行dfs()
if (can(r, j)){
p[r][j] = 1;            //将该位置设为有棋子
dfs(r+1);               //到下一行，调用dfs(r+1)
p[r][j] = 0;            //回溯，将该位置设为没有棋子
}
}
}
}
int main()
{
memset(p, 0, sizeof(p));
dfs(0);                             //从第一行开始搜索
printf("cnt = %d\n", cnt);
return 0;
}