POJ_1321（DFS算法）

POJ_1321，dfs不规则棋盘放棋子问题

#include <iostream>
#include <cstring>
using namespace std;
//chess该位置可否放棋子(0不可1可)，不可可能有本身不能放或者该位置已经放了棋子 
//haschess第0行表示该行是否有棋子，第1行表示该列是否有棋子 
bool chess[8][8],haschess[2][8];
int ans,length,chessnum;
//按行放棋子，当前棋子和可放起始行号从0编号 
void mydfs(int curchess,int currline){
	if(curchess==chessnum){
		ans++;
		return;
	}
	else{
		//可放棋子位置行号范围：可放起始行号到棋盘长度-(要放棋子总数-已放棋子数) 
		for(int i=currline;i<=length-chessnum+curchess;i++)
			for(int j=0;j<length;j++)
				if(chess[i][j] && !haschess[0][i] && !haschess[1][j]){
					chess[i][j]=0;	haschess[0][i]=1;	haschess[1][j]=1;
					mydfs(curchess+1,currline+1);
					chess[i][j]=1;	haschess[0][i]=0;	haschess[1][j]=0;
				}
		return;
	}
}

int main(){
	while(cin>>length>>chessnum){
		if(length==-1)	return 0;
		char canput;
		for(int i=0;i<length;i++)
			for(int j=0;j<length;j++){
				cin>>canput;
				if(canput=='#')	chess[i][j]=1;
				else			chess[i][j]=0;
			}
		ans=0;
		memset(haschess,0,sizeof(haschess));
		mydfs(0,0);
		cout<<ans<<endl;
	}
}