c语言使用dfs解决n皇后问题

#include <stdio.h>

#define NUM 4             //棋盘的大小 

int a[NUM][NUM];        //棋盘数组 
int count =0;             //解的个数

//dfs函数,i表示目前棋盘上有几个皇后,a表示棋盘数组 
void dfs(int i,int a[NUM][NUM]);

//在棋盘上的i,j,处放置皇后,如果满足条件返回1 
int isRight(int i,int j,int a[NUM][NUM]);

int main(int argc, char *argv[])
{
    int i,j;
    //初始化棋盘,0代表未放置皇后 
    for(i=0;i<NUM;i++){
        for(j=0;j<NUM;j++) a[i][j]=0;
    }
    //开始时,棋盘上没有皇后 
    dfs(0,a);
    //输出结果 
    printf("have %d answers\n",count);        
    return 0;
}

void dfs(int i,int a[NUM][NUM]){
    if(i==NUM){            //如果i==NUM,说明棋盘上以及放够了所需的皇后数目 
        count++;        //解数加一,并输出棋盘
        int n,m; 
        for(n=0;n<NUM;n++){
            for(m=0;m<NUM;m++) {printf("%d  ",a[n][m]);}
            printf("\n");
        } 
        
        printf("\n\n");
        return;            //必须返回,即搜索到达底部 
    } 
    
    //n皇后摆放时,必须在每一行都有一个 
    int k; 
    for(k=0;k<NUM;k++){            //第一行遍历 
        if(isRight(k,i,a)){        //判断是否符合要求 
            a[k][i]=1;            //如果符合要求,放入 
            dfs(i+1,a);            //向下探索 
            a[k][i]=0;            //如果执行到这里说明探索失败,不能放入 
        }
    }
}

//判断是否可以在i,j处向棋盘放置皇后 
int isRight(int i,int j,int a[NUM][NUM]){
    int x,y;
    for(x=i,y=0;y<NUM;y++){
        if(a[x][y]==1 && y!=j) return 0;
    } 
    for(x=0,y=j;x<NUM;x++){
        if(a[x][y]==1 && x!=i) return;
    }
    for(x=i-1,y=j-1;x>=0 && y>=0;x--,y--){
        if(a[x][y]==1 ) return 0;
    } 
    for(x=i+1,y=j+1;x<NUM && y<NUM;x++,y++){
        if(a[x][y]==1 ) return 0;
    }
    for(x=i-1,y=j+1;x>=0 && y<NUM;x--,y++){
        if(a[x][y]==1 ) return 0;
    }
    for(x=i+1,y=j-1;x<NUM && y>=0;x++,y--){
        if(a[x][y]==1 ) return 0;
    }
    
    return 1;
}