棋盘问题
Time Limit: 1000MS | Memory Limit: 10000K | |
Total Submissions: 18174 | Accepted: 8975 |
Description
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
每组数据的第一行是两个正整数,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
思路:类似于8皇后问题,但是棋子的数目与棋盘的行数不想等,不一定每行都能放上棋子,所以要枚举哪一行作为放棋子的第一行,显然第一行只能出现在第0——(n-k)行中。另外要用数组row、col分别记录某行和某列是否已放有棋子。
AC代码:
#include <cstring> #include <string> #include <cstdio> #include <algorithm> #include <queue> #include <cmath> #include <vector> #include <cstdlib> #include <iostream> #define max2(a,b) ((a) > (b) ? (a) : (b)) #define min2(a,b) ((a) < (b) ? (a) : (b)) using namespace std; int n,k; char map[10][10]; int ans; bool row[10],col[10]; void dfs(int cur,int num) { if(num==k) { ans++; return; } for(int i=cur; i<n; i++) for(int j=0; j<n; j++) { if(map[i][j]=='#'&&!row[i]&&!col[j]) { row[i]=true; col[j]=true; dfs(i,num+1); row[i]=false; col[j]=false; } } } int main() { while(scanf("%d%d",&n,&k)!=EOF) { if(n==-1&&k==-1) break; for(int i=0; i<n; i++) scanf("%s",map[i]); ans=0; memset(row,false,sizeof(row)); memset(col,false,sizeof(col)); int start=n-k; for(int i=0;i<=start;i++) //枚举那一行作为第一行 for(int j=0;j<n;j++) { if(map[i][j]=='#') { row[i]=true; col[j]=true; dfs(i+1,1); row[i]=false; col[j]=false; } } printf("%d\n",ans); } return 0; }