状态压缩。比较简单。
一开始每种状态只记录有无,方法数明显少了,改为记录到达此状态方法数后正确。
第一次碰到SGU的题,在文件输入输出上中枪了。。。单组数据,“Test #1”什么的不在输入输出内。被坑2WA。
然后数据范围超64位,先改了下sum,由WAT84变成T100,发现记录状态转移的数组也要64位
#include<cstdio>
#include<cstring>
#include<iostream>
#include<iostream>
using namespace std;
int n,k;
int num;
int s[505];
int c[505];
__int64 dp[20][505][505];
int Con(int x)
{
int ans=0;
while(x)
{
ans+=x&1;
x>>=1;
}
return ans;
}
int main()
{
scanf("%d %d",&n,&k);
int ptr=0;
__int64 sum=0;
for(int i=0;i<(1<<n);i++)
{
if((i<<1)&i)continue;
s[ptr]=i;
c[ptr++]=Con(i);
}
memset(dp,0,sizeof(dp));
for(int i=0;i<ptr;i++)
{
dp[0][i][c[i]]=1;
if(c[i]==k)sum++;
}
for(int i=1;i<n;i++)
{
for(int j=0;j<ptr;j++)
{
for(int l=0;l<ptr;l++)
{
if((s[j]&s[l])||(s[j]&(s[l]<<1))||(s[j]&(s[l]>>1)))continue;
for(int z=0;z+c[j]<=k;z++)
{
dp[i][j][z+c[j]]+=dp[i-1][l][z];
if(z+c[j]==k&&c[j]!=0){sum+=dp[i-1][l][z];}
}
}
}
}
printf("%I64d\n",sum);
return 0;
}