UVA - 1291 Dance Dance Revolution 多维dp

问题

https://vjudge.net/problem/UVA-1291

分析

dp[i][j][k]表示时间点在i时，左脚在j,右脚在k的状态的最小花费
使用了二维数组，作为代价矩阵，简化了迈步时的代价计算

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <map>
#include <string>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;
const int maxn=1000+5,Inf=0x3f3f3f3f;
int n,s,kase=0,dir[maxn],dp[maxn][5][5],value[5][5];

//初始化转移代价矩阵
void init(){
    for(int i=0;i<5;++i) value[i][i]=1;
    for(int i=1;i<5;++i) value[0][i]=value[i][0]=2;
    value[1][2]=value[2][3]=value[3][4]=value[1][4]=3;
    value[1][3]=value[2][4]=4;
    //对称矩阵
    for(int i=1;i<5;++i){
        for(int j=0;j<i;++j)
            value[i][j]=value[j][i];
    }
}

int main(void){
    init();
    while(scanf("%d",&dir[1])==1 && dir[1]!=0){
        for(n=2;;++n){
            scanf("%d",&dir[n]);
            if(dir[n]==0) break;
        }
        --n;
        memset(dp,Inf,sizeof(dp));
        dp[0][0][0]=0;
        //刷表法，用当前的状态去更新下一个时间的状态，开始要初始化
        for(int i=0;i<n;++i){
            for(int j=0;j<5;++j){
                for(int k=0;k<5;++k){
                    int &temp=dp[i][j][k];
                    if(temp==Inf) continue;
                    if(k!=dir[i+1]) dp[i+1][dir[i+1]][k]=min(dp[i+1][dir[i+1]][k],temp+value[j][dir[i+1]]);
                    if(j!=dir[i+1]) dp[i+1][j][dir[i+1]]=min(dp[i+1][j][dir[i+1]],temp+value[k][dir[i+1]]);
                }
            }
        }
        int ans=Inf;
        for(int j=0;j<5;++j){
            for(int k=0;k<5;++k){
                ans=min(ans,dp[n][j][k]);
            }
        }
        printf("%d\n",(ans==maxn)?0:ans);
    }
    return 0;
}