NYOJ 炫舞家st

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int INF=0x3f3f3f3f;
int dp[10003][5][5];
int a[10003];
int main()
{
int t[5][5];
t[0][1]=t[0][2]=t[0][3]=t[0][4]=2;
t[1][2]=t[2][1]=t[2][3]=t[3][2]=t[4][1]=t[1][4]=t[3][4]=t[4][3]=3;
t[2][4]=t[4][2]=t[3][1]=t[1][3]=4;
t[1][1]=t[2][2]=t[3][3]=t[4][4]=1;
memset(dp,INF,sizeof(dp));
dp[0][0][0]=0; 
int i; 
while(1)
{
for(i=1;;i++)
{
cin>>a[i];
if(!a[1])
break;
if(!a[i])
break;
for(int j=0;j<5;j++)
for(int k=0;k<5;k++)
{
dp[i][j][a[i]]=min(dp[i-1][j][k]+t[k][a[i]],dp[i][j][a[i]]);//右脚踩
dp[i][a[i]][j]=min(dp[i-1][k][j]+t[k][a[i]],dp[i][a[i]][j]);//左脚踩
}
}
int result=INF;
for(int j=0;j<5;j++)
for(int k=0;k<5;k++)
result=min(result,dp[i-1][j][k]);
cout<<result<<endl;
}
return 0;
}