c语言dfs解决n皇后问题

原文链接: c语言dfs解决n皇后问题

上一篇: c语言链表栈实现进制转换

下一篇: c语言dfs解决全排列问题

问题描述:

在n×n格的棋盘上放置彼此不受攻击的n个皇后。按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。n后问题等价于再n×n的棋盘上放置n个皇后，任何2个皇后不妨在同一行或同一列或同一斜线上

简单实现

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