递推DP URAL 1586 Threeprime Numbers

题目传送门

 /*
     题意：n位数字，任意连续的三位数字组成的数字是素数，这样的n位数有多少个
     最优子结构：考虑3位数的数字，可以枚举出来，第4位是和第3位，第2位组成的数字判断是否是素数
                 所以，dp[i][j][k] 表示i位数字，最高位数字j，第二高位数字k
                 状态转移方程：dp[i][j][k] += dp[i-1][k][l]
                 注意：最高位从1开始枚举:)
     详细解释：http://blog.csdn.net/zhangyanxing666/article/details/9628563
 */
 #include <cstdio>
 #include <iostream>
 #include <algorithm>
 #include <cstring>
 #include <cmath>
 using namespace std;

 const int MAXN = 1e4 + ;
 const int INF = 0x3f3f3f3f;
 const int MOD = 1e9 + ;
 int prime[][][];
 int vis[];
 int dp[MAXN][][];

 void solve(void)
 {
     memset (prime, , sizeof (prime));
     memset (vis, , sizeof (vis));
     memset (dp, , sizeof (dp));
     for (int i=; i<=; ++i)
     {
         if (!vis[i])
         {
             vis[i] = true;
             for (int j=i*; j<=; j+=i)
             {
                 vis[j] = true;
             }
             prime[i/][i/%][i%] = ;
         }
     }

     for (int i=; i<=; ++i)
     {
         for (int j=; j<=; ++j)
         {
             for (int k=; k<=; ++k)    if (prime[i][j][k])    dp[][i][j]++;
         }
     }

     for (int i=; i<=; ++i)
     {
         for (int j=; j<=; ++j)
         {
             for (int k=; k<=; ++k)
             {
                 for (int l=; l<=; ++l)
                     if (prime[j][k][l])    dp[i][j][k] = (dp[i][j][k] + dp[i-][k][l]) % MOD;
             }
         }
     }
 }

 int main(void)        //URAL 1586 Threeprime Numbers
 {
     //freopen ("M.in", "r", stdin);

     solve ();
     int n;
     while (scanf ("%d", &n) == )
     {
         int ans = ;
         for (int i=; i<=; ++i)
         {
             for (int j=; j<=; ++j)
                 ans = (ans + dp[n][i][j]) % MOD;
         }

         printf ("%d\n", ans);
     }

     return ;
 }