题目地址:http://livearchive.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=62&page=show_problem&problem=32
/*
Author:Bob Lee
2012.8.25
====================================================
这题是用DP来解决策略选择的问题
解法在LRJ的黑书上说的很详细了
只是把自己的体会再说说
LRJ给出的状态方程如下
dp[i][j][k] = min(dp[data[k]][j][k+1] + effort(i,data[k]) ,
dp[i][data[k]][k+1] + effort(j,data[k]));
i,j表示的是左右脚的位置,k表示的第k个舞步
你会发现后面的状态转移方程怎么是从k+1到k啊
我想了一会,由此联想到了在HDU上做的一道数塔的题目。
你可以正着DP,但是到最后的时候你要把最后同一级别的状态枚举,这样才能找出最优解。
但是如果你倒着解,最优解就是第一个,因为状态是越来越多的,但是初始状态是一定的。
再反过来想这个问题,就理解了。
*/
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 10010
int dp[5][5][MAXN];
int n;
int data[MAXN];
int effort(int a,int b)
{
if(a == b)
return 1;
if(a == 0)
return 2;
if(abs(a-b) == 2)
return 4;
return 3;
}
int read()
{
n=0;
int a;
while(scanf("%d",&a) && a != 0)
{
data[++n] = a;
}
return n;
}
//这个有点像数塔的意思,为了如果是正着求得话,后面会要枚举,那很麻烦
void init()
{
int i,j;
for(i=0;i<5;i++)
for(j=0;j<5;j++)
dp[i][j][n+1] = 0;
}
int dp_s()
{
int i,j,k;
for(k=n;k>0;k--)
{
for(i=0;i<5;i++)
{
for(j=0;j<5;j++)
{
dp[i][j][k] = min(dp[data[k]][j][k+1] + effort(i,data[k]) ,
dp[i][data[k]][k+1] + effort(j,data[k]));
}
}
}
return dp[0][0][1];
}
int main()
{
while(read())
{
init();
printf("%d\n",dp_s());
}
return 0;
}
上面的是普通版本的,由于每次求解只是依赖与上一步是怎么求的,所以我们可以将滚动数组的概念加进来,这样就可以省去大量的空间。下面我贴上滚动数组版本的源代码,其实就修改来一个地方,但是时间为我升了0.1s,竟然挤到UVAlive的首页第7!!!这不科学。。。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 10010
int dp[5][5][2];//由于使用来滚动数组,空间昼减
int n;
int data[MAXN];
int effort(int a,int b)
{
if(a == b)
return 1;
if(a == 0)
return 2;
if(abs(a-b) == 2)
return 4;
return 3;
}
int read()
{
n=0;
int a;
while(scanf("%d",&a) && a != 0)
{
data[++n] = a;
}
return n;
}
//这个有点像数塔的意思,为了如果是正着求得话,后面会要枚举,那很麻烦
void init()
{
int i,j;
for(i=0;i<5;i++)
for(j=0;j<5;j++)
dp[i][j][(n+1)%2] = 0;//这里就是滚动数组修改的地方
}
int dp_s()
{
int i,j,k;
for(k=n;k>0;k--)
{
for(i=0;i<5;i++)
{
for(j=0;j<5;j++)
{
dp[i][j][k%2] = min(dp[data[k]][j][(k+1)%2] + effort(i,data[k]) ,
dp[i][data[k]][(k+1)%2] + effort(j,data[k]));//利用来滚动数组
}
}
}
return dp[0][0][1];
}
int main()
{
while(read())
{
init();
printf("%d\n",dp_s());
}
return 0;
}