POJ 1321棋盤問題 （dfs）

棋盤問題

在一個給定形狀的棋盤（形狀可能是不規則的）上面擺放棋子，棋子沒有差別。要求擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列，請程式設計求解對于給定形狀和大小的棋盤，擺放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，是以比平常的棋盤問題要多一些判斷，但主要還是深搜dfs。

AC代碼

#include <stdio.h>
#include <string.h>

int vis[1][18], n, k, cnt = 0;
char s[9][9];
void dfs(int cur, int num) {
	if(num == k) {
		cnt++;
		return ;
	}
	for(int i = 0; i < n; i++) {
		if(s[cur][i] == '#' && !vis[0][i]) {
			vis[0][i] = 1;
			dfs(cur + 1, num + 1);
			vis[0][i] = 0;
		}
	}
	if(k - num <= n - cur - 1) dfs(cur + 1, num);
}
int main() {
	while(scanf("%d %d", &n, &k) && (n != -1 || k != -1)) {
		memset(vis, 0, sizeof(vis));
		for(int i = 0; i < n; i++) scanf("%s", s[i]);
		for(int i = 0; i < n; i++) {
			if(s[0][i] == '#') {
				vis[0][i] = 1;
				dfs(1, 1);
				vis[0][i] = 0;
			}
		}
		if(k != n) dfs(1, 0);
		printf("%d\n", cnt);
		cnt = 0;
	}
	return 0;
}
           

End