uva1291 Dance Dance Revolution(dp)

黑书1.5.1例题4.舞蹈家怀特先生。
dp[i][x][y] 表示跳完第i个舞步，左脚在x，右脚在y所需花费的最小体力。状态转移方程：

$$dp[i][x][a[i]]=min(dp[i][x][a[i]],dp[i-1][x][y]+move(y,a[i]))$$

$$dp[i][a[i]][y]=min(dp[i][a[i]][y],dp[i-1][x][y]+move(x,a[i]))$$
一开始dp数组都给inf，只有第0次站在0,0给0，这样就避免了不合法的决策产生。

#include <cstdio>
#include <cstring>
#define N 10100
#define inf 0x3f3f3f3f
int a[N],n,dp[N][5][5],ans;
template<typename T>
inline T min(T x,T y){return x<y?x:y;}
inline int swap(int &x,int &y){x^=y;y^=x;x^=y;}
inline int move(int x,int y){
    if(x==y) return 1;
    if(x<y) swap(x,y);
    if(y==0) return 2;
    if(x-y==2) return 4;
    else return 3;
}
int main(){
//  freopen("a.in","r",stdin);
    while(1){
        n=0;ans=inf;
        while(1){
            scanf("%d",&a[++n]);
            if(a[n]==0) break;
        }
        n--;if(n==0) break;
        memset(dp,0x3f,sizeof(dp));
        dp[0][0][0]=0;
        for(int i=1;i<=n;++i)
            for(int x=0;x<=4;++x)
                for(int y=0;y<=4;++y)
                    dp[i][x][a[i]]=min(dp[i][x][a[i]],dp[i-1][x][y]+move(y,a[i])),
                    dp[i][a[i]][y]=min(dp[i][a[i]][y],dp[i-1][x][y]+move(x,a[i]));
        for(int i=0;i<=4;++i)
            ans=min(ans,min(dp[n][i][a[n]],dp[n][a[n]][i]));
        printf("%d\n",ans);
    }
    return 0;
}