【HDOJ】1438 钥匙计数之一

状态压缩。分最后一个槽的值以及当前的配置方案是否可以进行DP。

 /* 1438 */
 #include <cstdio>
 #include <cstring>
 #include <cstdlib>

 #define MAXN 32

 const int MAXS = <<;

 __int64 dp[MAXN][MAXS][][];
 int cnt[<<];

 int abs(int x) {
     return x< ? -x:x;
 }

 int main() {
     int i, j, k, n;
     int s;
     __int64 ans;

     memset(cnt, , sizeof(cnt));
     for (i=; i<MAXS; ++i)
         for (j=; j<; ++j)
             if (i & (<<j))
                 ++cnt[i];

     memset(dp, , sizeof(dp));
     for (i=; i<; ++i)
         dp[][<<i][i][] = ;

     for (n=; n<MAXN; ++n) {
         for (i=; i<MAXS; ++i) {
             for (j=; j<; ++j) {
                 for (k=; k<; ++k) {
                     s = i | (<<j);
                     dp[n][s][j][] += dp[n-][i][k][];
                     if (abs(j-k) == ) {
                         dp[n][s][j][] += dp[n-][i][k][];
                     } else {
                         dp[n][s][j][] += dp[n-][i][k][];
                     }
                 }
             }
         }
     }

     for (n=; n<MAXN; ++n) {
         ans = ;
         for (i=; i<MAXS; ++i) {
             if (cnt[i] >= )
                 ans += dp[n][i][][] + dp[n][i][][] +\
                      dp[n][i][][] + dp[n][i][][];
         }
         printf("N=%d: %I64d\n", n, ans);
     }

     return ;
 }