首页 > 代码库 > poj 1321

poj 1321

A - 棋盘问题
Time Limit:1000MS     Memory Limit:10000KB     64bit IO Format:%I64d & %I64u
Submit Status Practice POJ 1321

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

21

 

 k==n的情况很简单,主要考虑k<n的情况。
dfs(r,sum)  摆到第r行,用了多少个棋子。
#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#include<string>#include<cmath>#include<algorithm>#include<queue>#include<vector>#include<set>using namespace std;int n,k,ans;char s[10][10];bool vis[10][10],cal[10];void dfs(int r,int sum){      if(sum==k)      {            ans++;            return ;      }      if(r>n)            return ;      for(int j=1;j<=n;j++)            if(vis[r][j]&&!cal[j])            {                  cal[j]=1;                  dfs(r+1,sum+1);                  cal[j]=0;            }      dfs(r+1,sum);//这里注意,k<n时,跳过r行,直接搜r+1行。      return ;}int main(){      while(scanf("%d%d",&n,&k)!=EOF)      {            ans=0;            memset(vis,0,sizeof(vis));            memset(cal,0,sizeof(cal));            if(n==-1&&k==-1)                  break;            for(int i=1;i<=n;i++)            {                 getchar();                 for(int j=1;j<=n;j++)                 {                        scanf("%c",&s[i][j]);                        if(s[i][j]==‘#‘)                             vis[i][j]=1;                 }            }            dfs(1,0);            printf("%d\n",ans);      }      return 0;}

  

poj 1321