UVA 1291 Dance Dance Revolution | dp

对于dp，做了一定的题量之后就可以开始YY了。

题意：

一个跳舞机，图应该是这样的吧。官方的图挂了。

在这些点上移动脚步所消耗的能量不尽相同。给出一个跳舞的序列，然后让你求出完成这个序列所消耗的最少能量。两只脚不能同时在一个点上。

思路：

状态定义：dp[i][j][z]：第i步，左脚移动到j，右脚移动到z的最少消耗的能量是多少。

状态转移：如果第 i 步所要进入的板块已经有脚存在，则 dp[i][j][z] = min(dp[i][j][z], dp[i-1][j][z] + 1);

如果你决定第 i 步打算用左脚去踩，则dp[i][t][z] = min(dp[i][t][z], dp[i-1][j][z] + j 块移动到t 块的花费);//t 是第i步所要进入的格子。

如果决定用右脚去踩，则dp[i][j][t] = min(dp[i][j][t], dp[i-1][j][z] + z 块移动到t 块的花费);

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <vector>
const int MAXN = 100005;
int dp[MAXN][6][6];
using namespace std;
vector <int> seq;
int cos[5][5];
const int INF = 0x3f3f3f3f;
void buildcos()
{
    for(int i = 0;i < 5; i++)
    {
        cos[0][i] = cos[i][0] = 2;
    }
    for(int i = 1;i <= 4; i++)
        cos[i][i] = 1;
    cos[1][3] = cos[3][1] = 4;
    cos[2][4] = cos[4][2] = 4;
    for(int i = 1;i <= 4; i++)
    {
        if(i+1 == 5)    cos[i][1] = cos[1][i] = 3;
        else    cos[i][i+1] = cos[i+1][i] = 3;
    }
}
int solve()
{
    int n = seq.size();
    memset(dp, INF, sizeof(dp));
    dp[0][0][0] = 0;
    for(int i = 1;i <= n; i++)
    {
        int& t = seq[i-1];
        for(int j = 0;j <= 4; j++)
        {
            for(int z = 0;z <= 4; z++)
            {
                //left
                if(dp[i-1][j][z] != INF)
                {
                    if(t == j || t == z)  dp[i][j][z] = min(dp[i][j][z], dp[i-1][j][z] + 1);
                    //left
                    dp[i][t][z] = min(dp[i][t][z], dp[i-1][j][z] + cos[t][j]);
                    //right
                    dp[i][j][t] = min(dp[i][j][t], dp[i-1][j][z] + cos[t][z]);
                }
            }
        }
    }
    int& t = seq[seq.size() - 1];
    int res = INF;
    for(int i = 1;i <= 4; i++)
    {
        res = min(res, dp[n][i][t]);
        res = min(res, dp[n][t][i]);
    }
    return res;
}


int main()
{
    while(1)
    {
        int t;
        seq.clear();
        while(scanf("%d", &t), t)
        {
            seq.push_back(t);
        }
        if(seq.empty())    break;
        //
        buildcos();
        int res = solve();
        cout<<res<<endl;
    }
    return 0;
}