1. 程式人生 > >POJ 1321 chessboard-problem

POJ 1321 chessboard-problem

[Description]:

在一個給定形狀的棋盤(形狀可能是不規則的)上面擺放棋子,棋子沒有區別。要求擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列,請程式設計求解對於給定形狀和大小的棋盤,擺放k個棋子的所有可行的擺放方案C。 [Input]: 輸入含有多組測試資料。 
每組資料的第一行是兩個正整數,n k,用一個空格隔開,表示了將在一個n*n的矩陣內描述棋盤,以及擺放棋子的數目。 n <= 8 , k <= n 
當為-1 -1時表示輸入結束。 
隨後的n行描述了棋盤的形狀:每行有n個字元,其中 # 表示棋盤區域, . 表示空白區域(資料保證不出現多餘的空白行或者空白列)。 
[Output
]:

對於每一組資料,給出一行輸出,輸出擺放的方案數目C (資料保證C<2^31)。

[Sample Input]:

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1
[Sample Output]:
2
1

[解題思路]:對於這道題大多數用的是進行層層深搜的思想,用一個數組來記錄每列的狀態,需要中間判斷下放不放子的問題就行了,資料量因為n<=8,所以不存在什麼問題。

source_code:

/*
 * =====================================================================================
 *
 *       Filename:  chessboard.cpp
 *        Version:  1.0
 *        Created:  2013年10月13日 16時45分23秒
 *       Revision:  none
 *       Compiler:  clang
 *
 * =====================================================================================
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
int chessboard[9][9];
int n,souce_way;
int flag_tod[9];
void find_souce(int r,int k){
	int i;
	if(k==0){
	      ++souce_way;
	      return ;
	}
	if(r>=n)
	      return ;
	for(i=0;i<n;++i){
		if(chessboard[r][i]&&!flag_tod[i]){
			flag_tod[i]=1;
			find_souce(r+1,k-1);
			flag_tod[i]=0;
		}
	}
	find_souce(r+1,k);
}
int main(int argc, char *argv[])
{
	int k,i,j;
	char ch;
	while(1){
		printf("Test start ,please set value (n k):  ");
		scanf("%d %d",&n,&k);
		getchar();
		if(n==-1&&k==-1)
		      break;
		for(i=0;i<n;++i){
			for(j=0;j<n;++j){
				ch=getchar();
				if(ch=='#')
				      chessboard[i][j]=1;
				else
				      chessboard[i][j]=0;
			}
			getchar();
		}
		for(i=0;i<9;i++)
		      flag_tod[i]=0;
		souce_way=0;
		find_souce(0,k);
		printf("ways: %d \n",souce_way);
	}
	return EXIT_SUCCESS;
}