ACM: 动态规划 黑书上的题目

                                                                                                                      "跳舞机" 怀特先生
Description
怀特先生是一个大胖子。他很喜欢玩跳舞机（Dance Dance Revolution, DDR），甚至希望有一天人家会脚踏“舞蹈家怀特先生”。可惜现在他的动作根本不能称作是在跳舞，尽管每次他都十分投入的表演。这也难怪，有他这样的体型，玩跳舞机是相当费劲的。因此，他希望写一个程序来安排舞步，让他跳起来轻松一些，至少不要每次都汗流浃背。
DDR的主要内容是用脚来踩踏板。 踏板有四个方向的箭头，用1 (Up)、2 (Left)、3 (Down)、4 (Right)来代表，中间位置由0来代表。每首歌曲有一个箭头序列，游戏者必须按照或这个序列一次用某一只脚踩相应的踏板。在任何时候，两只脚都不能在同一踏板上，但可以同时待在中心位置0。
每一个时刻，它必须移动而且只能移动他的一只脚去踩相应的箭头，而另一只脚不许移动。这样，它跳DDR的方式可以用遗传数字l1, l2, ..., ln来表示，其中li=0表示他用左脚踩第i个箭头，li=1表示他用右脚踩第i个箭头。
跳完一首曲子之后，怀特先生会计算他所消耗的体力。从中心移动到任何一个箭头耗费2单位体力，从任何一个箭头移动到相邻箭头耗费3单位体力，移动到相对的箭头（1和3相对，2和4相对）耗费4单位体力，而留在原地再踩一下只需要1单位。怀特先生应该怎样移动他的双脚（即，对于每个箭头，选一只脚去踩它），才能用最少的体力完成一首给定的舞曲呢？
例如，对于箭头序列Left (2), Left (2), Up (1), Right (4)，他应该分别用左、左、右、右脚去踩，总的体力耗费为2+1+2+3=8单位.

Input
4
2 2 4 4
Output
8

题意:  跳舞机要求最后消耗最少的体力.

解题思路: 黑书上的题目.
                            1. 决策: (1)把一只脚从中心位置移到目标位置.
                                                      (2)一只脚移动到相邻的位置.
                                                      (3)一只脚移动到相对的位置.
                                                      (4)一只脚再原地再踩一下.
                            2. 状态: dp[x][y][k]: 跳到第k步时, 左脚在x的位置, 右脚在y的位置, 最少的体力消耗.
                            3. 状态转移方程: dp[x][y][k] = min(dp[a[k]][y][k+1]+effort[x][a[k]] , dp[x][a[k]][k+1]+effort[y][a[k]]);
                                                                                  a[k]: 表示第k步的舞步是要到的位置, effort[i][j]: 表示从i位置到j位置的体力消耗.

代码:
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
#define MAX 105

int effort[5][5] = {
                  {1, 2, 2, 2, 2},
                  {2, 1, 3, 4, 3},
                  {2, 3, 1, 3, 4},
                  {2, 4, 3, 1, 3},
                  {2, 3, 4, 3, 1}
                };

int n;
int a[MAX]; //0:原地, 1:上  3:下  4:左  2:右
int dp[5][5][MAX];

inline int min(int a,int b)
{
      return a < b ? a : b;
}

int solve(int x,int y,int k)
{
      if(dp[x][y][k] != -1)
            return dp[x][y][k];
      int ans = 0;
      if(k <= n)
      {
            ans = min( solve(a[k],y,k+1)+effort[x][a[k]] , solve(x,a[k],k+1)+effort[y][a[k]] );
      }
      dp[x][y][k] = ans;
      return ans;
}

int main()
{
//      freopen("input.txt","r",stdin);
      while(scanf("%d",&n) != EOF)
      {
            for(int i = 0; i < n; ++i)
                  scanf("%d",&a[i]);
            memset(dp,-1,sizeof(dp));
           
            int result = solve(0,0,0);
            printf("%d\n",result);
      }
      return 0;
}