poj 1321 棋盘问题
题目链接:http://poj.org/problem?id=1321
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
直接进行深搜dfs,找到一种情况就加一次,同时注意回溯。
代码如下:
1 #include<stdio.h> 2 #include<string.h> 3 int n,k,sum; 4 int str[9][9]; 5 int vx[9],vy[9];//标记数组,vx[i]标记用过的行,vy[i]标记用过的列 6 void dfs(int x,int cur) 7 { 8 int i,j; 9 //if(x>n) return; 10 if(cur==k)//如果找齐K个就返回找其他的情况 11 { 12 sum++; 13 return ; 14 } 15 16 for(i=x;i<=n;i++) //从x行找起走 17 { 18 for(j=1;j<=n;j++) 19 { 20 if(str[i][j]=='#') 21 { 22 if(!vx[i]&&!vy[j]) 23 { 24 vx[i]=1; 25 vy[j]=1; 26 dfs(i+1,cur+1); 27 vx[i]=0; //开始回溯 28 vy[j]=0; 29 } 30 } 31 } 32 } 33 return ; 34 } 35 int main() 36 { 37 int i,j; 38 while(scanf("%d%d",&n,&k)!=EOF) 39 { 40 getchar(); 41 if(n==-1&&k==-1) break; 42 for(i=1;i<=n;i++) 43 { 44 for(j=1;j<=n;j++) 45 { 46 scanf("%c",&str[i][j]); 47 } 48 getchar(); 49 } 50 sum=0; 51 52 for(i=1;i<=n-k+1;i++) //自上往下找,这要就能避免找到重复的情况,而且K个元素,至少要k行 53 { 54 for(j=1;j<=n;j++) 55 { 56 memset(vx,0,sizeof(vx));//找第一个可放的位置时要把行和列重归0 57 memset(vy,0,sizeof(vy)); 58 if(str[i][j]=='#')//找到第一个可以放的位置 59 { 60 if(!vx[i]&&!vy[j]) 61 { 62 vx[i]=1; 63 vy[j]=1; 64 dfs(i+1,1); 65 } 66 } 67 } 68 } 69 printf("%d\n",sum); 70 } 71 return 0; 72 }