HIT 2662 Pieces Assignment

来源:互联网 发布:github ssh项目源码 编辑:程序博客网 时间:2024/05/17 04:54

Background

    有一个n*m的棋盘(n、m≤80,n*m≤80)要在棋盘上放k(k≤20)个棋子,使得任意两个棋子不相邻(每个棋子最多和周围4个棋子相邻)。求合法的方案总数。

Input

    本题有多组测试数据,每组输入包含三个正整数n,m和k。

Output

    对于每组输入,输出只有一个正整数,即合法的方案数。

Sample Input

2 2 34 4 1

Sample Output

0

16

挺基础的一道状压dp啦,然而瞄了一眼题解就没有开long long,wa了一中午。

然而题解开int过了,非酋表示不服。


#include<algorithm>#include<iostream>#include<cstring>#include<cstdio>using namespace std;int n,m,p,tp,a[1005],b[1005];long long ans,f[85][25][1005];void dfs(int num,int lst,int now,int cnt){if(num==m){++tp;a[tp]=now;b[tp]=cnt;return ;}dfs(num+1,lst,now,cnt);//不选 if(num-lst>=2&&cnt<p)//选 dfs(num+1,num,now|(1<<num),cnt+1);}int main(){while(scanf("%d%d%d",&n,&m,&p)==3){tp=ans=0;memset(a,0,sizeof(a));memset(b,0,sizeof(b));memset(f,0,sizeof(f));if(n<m)swap(n,m);dfs(0,-2,0,0);f[0][0][0]=1;for(int i=1;i<=n;i++)for(int j=1;j<=tp;j++)//上一行摆放状态 for(int g=0;g<=p;g++)//前(i-1)行的棋子总数 if(f[i-1][g][a[j]]>0)for(int k=1;k<=tp;k++)if((a[j]&a[k])==0&&g+b[k]<=p)f[i][g+b[k]][a[k]]+=f[i-1][g][a[j]];for(int i=1;i<=tp;i++)ans+=f[n][p][a[i]];printf("%lld\n",ans);}return 0;}


0 1