poj 1321 棋盘问题 状压dp

来源:互联网 发布:zabbix snmp 端口号 编辑:程序博客网 时间:2024/06/18 06:10

棋盘问题
Time Limit: 1000MS Memory Limit: 10000K
Total Submissions: 46153 Accepted: 22302
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

居然在poj看到了中文题23333
其实是一道爆搜题,然而用状压做就更快啦。

对棋子的总数是有限制的,所以我们需要统计最后符合棋子总数的状态。(统计每个状态有多少个棋子)
dp[i][j]表示前i行总共放j个棋子的总个数。
还是判断该种转移是否合法,合法就转移。

int q=j|(1<<f-1);

| 按位或,如果两个数有一个数二进制的某一位是1,那么返回值的二进制的该位就会是1.所以可以用来统计这两种状态合并后总共有多少个1。

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>using namespace std;const int N = 10;const int M = 1<<10;int n,k;int num[M];int mp[N][N];int dp[N][M];int count(int x){    int cnt=0;    while(x){        cnt++;        x&=(x-1);    }    return cnt;}#define ms(x,y) memset(x,y,sizeof(x))void update(){    ms(dp,0);ms(mp,0);}int main(){    for(int i=0;i<(1<<10);i++) num[i]=count(i);    while(1){       scanf("%d%d",&n,&k);       if(n+k==-2) break;       update();       for(int i=1;i<=n;i++){          char s[10];          scanf("%s",s);          for(int j=1;j<=n;j++){              if(s[j-1]=='#') mp[i][j]=1;              else mp[i][j]=0;          }       }       dp[1][0]=1;       for(int i=1;i<=n;i++)          if(mp[1][i])             dp[1][1<<(i-1)]=1;       for(int i=2;i<=n;i++){          for(int j=0;j<(1<<n);j++){             if(num[j]>k) continue;//优化              dp[i][j]+=dp[i-1][j];             for(int f=1;f<=n;f++){                if(mp[i][f]&&(j&(1<<f-1))==0){//如果这个点可以放而且对j这种放法之前没有放过                     int q=j|(1<<f-1);                    dp[i][q] += dp[i-1][j];                }             }          }       }       int ans=0;       for(int i=0;i<(1<<n);i++){          if(num[i]==k)            ans+=dp[n][i];       }       printf("%d\n",ans);    }    return 0;}
原创粉丝点击