钥匙计数之一 - HDU 1438(状态压缩打表)

时间:2023-03-09 12:59:19
钥匙计数之一 - HDU 1438(状态压缩打表)

分析:首先想到每个钥匙的结尾有4种状态,不过题目还需要判断有三种不同的钥匙深度,所以每种深度结尾后有2^4种状态,0000->1111,不过题目还需需要有相邻的钥匙深度大于等于3,所以需要两种不同的状态表示0表示没有出现过这样的,1表示出现过,然后开一个4维的数组进行状态转移就好了dp[MAXN][bit][1<<bit][2]。

代码如下:

============================================================================================================================

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std; const int MAXN = ;
const int bit = ; long long dp[MAXN][bit][<<bit][]; int main()
{
dp[][][][] = dp[][][][] = ;
dp[][][][] = dp[][][][] = ; for(int i=; i<MAXN; i++)
for(int j=; j<bit; j++)
for(int k=; k<(<<bit); k++)
{
if(dp[i][j][k][])
{
for(int t=; t<; t++)
{
if(abs(j-t) == )
dp[i+][t][k|(<<t)][] += dp[i][j][k][];
else
dp[i+][t][k|(<<t)][] += dp[i][j][k][];
}
}
if(dp[i][j][k][])
{
for(int t=; t<; t++)
dp[i+][t][k|(<<t)][] += dp[i][j][k][];
}
} long long sum[MAXN]={}; for(int i=; i<MAXN; i++)
for(int j=; j<bit; j++)
for(int k=; k<; k++)
{
if(k==||k==||k>=)
sum[i] += dp[i][j][k][];
} for(int i=; i<MAXN; i++)
{
printf("N=%d: %lld\n", i, sum[i]);
} return ;
}