POJ 1321 棋盘问题 搜索

棋盘问题

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 19843		Accepted: 9888

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

AC代码

#include <iostream>
using namespace std;
#define MAX 10
char chess[MAX][MAX];
int count;
bool vis[MAX];
int n, k;

void dfs(int row, int num)//row表示棋盘的行数  num已经放入的棋子数
{
	if(k == num)
	{
		count++;
		return ;
	}
	if(row > n)
		return ;
	for(int j = 0; j < n; j++)
	{
		if(chess[row][j] == '#' && !vis[j])
		{
			vis[j] = 1;
			dfs(row+1, num+1);
			vis[j] = 0;
		}
	}
	dfs(row+1, num);//表示该行没有放置棋子 棋子数可能小于棋盘的行数
}

int main()
{
	while(cin >> n >> k)
	{	
		if(n == -1 && k == -1)
			return -1;
		memset(vis, 0, sizeof(vis));
		for(int i = 0; i < n; i++)
		{
			for(int j = 0; j < n; j++)
			{
				cin >> chess[i][j];
			}
			//getchar();
		}
		count = 0;
		dfs(0, 0);
		cout << count << endl;
	}
	return 0;
}
TLE代码

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
 
const int size = 10;
char map[size][size];
bool vis[size];
int n,
    k,
	cnt;

void dfs(int row, int num)
{
	if(num == k)
	{
		cnt++;
		return ;
	}
	for(int i = row; i < n; i++)
	{
		for(int j = 0; j < n; j++)
		{
 			if(map[i][j] == '#' && !vis[j])
			{   //每当找到一个‘#’ 对列判重 对行进行加1操作 满足题目中不在同一行或同一列的要求
				vis[j] = true;
				dfs(row + 1, num + 1);
				vis[j] = false;
			}
		}
	}
}

int main()
{
	while(scanf("%d%d", &n, &k))
	{
		if(n == -1 && k == -1)
			return -1;
		cnt = 0;
		memset(vis, false, sizeof(vis));
		int i, j;
		for(i = 0; i < n; i++)
		{
			getchar();//吃掉回车
			for(j = 0; j < n; j++)
			{
				map[i][j] = getchar();
			}
		}
		dfs(0, 0);//行 棋子数目
		printf("%d\n", cnt);
	}
	return 0;
}