棋盤問題 (dfs) POJ - 1321

題目大意：給一個N*N的區域，和M個棋子，詢問最後有多少種擺的方法，其中每個棋子不能放在同一行和同一列。

思路：對每一行進行dfs，dfs時使用book數組（雙重標記）標記。

#include<stdio.h>
#include<string.h> 
#define N 9
char z[N][N];//開空間存儲棋盤
int num;//計算總數
int bookh[N];//標記橫行是否有過棋子
int bookl[N];//標記縱行是否有過棋子
int n,k;
void dfs(int a,int m);
int main(void)
{
	
	while(scanf("%d%d",&n,&k),n!=-1){
		memset(bookh,0,sizeof(bookh));
		memset(bookl,0,sizeof(bookl));
		num=0;
		getchar();
		int i;
		for(i=0;i<n;i++) gets(z[i]);
		for(i=0;i<=n-k;i++){//優化，因爲只有前n-k行，才能存下k個棋子
			dfs(i,k);//dfs每一行
		}
		printf("%d\n",num);
	}
	return 0;
}
void dfs(int a,int m){
	if(m==0) {
//		num++;
		return ;
	}
	int i;
	for(i=0;i<n;i++){
		if(z[a][i]=='#'&&bookh[a]==0&&bookl[i]==0){//找到這一列中可以放置棋子的點
			if(m==1) {
				num++;
			//	return ;
			}
			bookh[a]=1;
			bookl[i]=1;
			int j;                 
			for(j=a+1;j<=n-m+1;j++){//尋找下一行來放置棋子
				dfs(j,m-1);
			}
			bookh[a]=0;
			bookl[i]=0;
		}
	}
	return ;
}