钥匙计数之一 (HDU-1438)（状压DP）

一把锁匙有N个槽，槽深为1，2，3，4。每锁匙至少有3个不同的深度且至少有1对相连的槽其深度之差为3。求这样的锁匙的总数。

Input

本题无输入

Output

对N>=2且N<=31，输出满足要求的锁匙的总数。

Sample Output

N=2: 0
N=3: 8
N=4: 64
N=5: 360
..
..
..
..
..
..
..

N=31: ...

注：根据Pku Judge Online 1351 Number of Locks或 Xi'an 2002 改编,在那里N<=16

题意：中文题，不过多说题意。

思路：这道题的话，因为每个槽有4种深度，所以一共有2^4种状态。然后开4维来保存每一次的状态：dp[ 第几个槽 ][ 当前状态 ][ 末尾深度 ][ 是否符合要求 ]即可。

AC代码：

#include <bits/stdc++.h>
typedef long long ll;
const int maxx=10010;
const int inf=0x3f3f3f3f;
using namespace std;
ll dp[32][1<<4+5][4][2];//第几个槽，槽的状态，最后一个槽的深度，是否已经符合要求
int num[20];
int main()
{
    memset(dp,0,sizeof(dp));
    memset(num,0,sizeof(num));
    for(int i=0; i<16; i++)
    {
        for(int j=0; j<4; j++)
        {
            if(i&(1<<j))
                num[i]++;//计算满足深度大于等于3
        }
    }
    dp[1][1][0][0]=dp[1][2][1][0]=dp[1][4][2][0]=dp[1][8][3][0]=1;
    for(int i=2; i<=31; i++)//第几个槽
    {
        for(int j=0; j<16; j++)//上一个槽状态
        {
            for(int k=0; k<4; k++)//上一个槽末尾
            {
                for(int l=0; l<4; l++)//这个槽末尾
                {
                    int ans=j|(1<<l);//当前状态
                    dp[i][ans][l][1]+=dp[i-1][j][k][1];
                    if(k-l==3 || l-k==3)
                    {
                        dp[i][ans][l][1]+=dp[i-1][j][k][0];
                    }
                    else
                    {
                        dp[i][ans][l][0]+=dp[i-1][j][k][0];
                    }
                }
            }
        }
    }
    ll cnt;
    for(int i=2; i<=31; i++)
    {
        cnt=0;
        for(int j=0; j<16; j++)
        {
            if(num[j]>=3)
            {
                for(int k=0; k<4; k++)
                {
                    cnt+=dp[i][j][k][1];
                }
            }
        }
        printf("N=%d: %lld\n",i,cnt);
    }
    return 0;
}