动态规划的解题步骤
- 定义状态(其实也就是描述最优子结构)
- 描述不同状态如何转移(一般可以求出一个状态转移方程)
- 按一个方向(一般都是自底而上)求出该问题的解
以01背包问题为例:
- 定义状态
定义状态dp[i][j]为前i件物品恰放入容量为j的背包时的最大价值;那么dp[n][v]就是这个问题的最终解; - 描述不同状态如何转移:
在这个过程中,我们很自然想到的决策就是第i件物品放还是不放;放与不放会直接影响到dp[i][j]的值;
a.放,那么dp[i][j]的状态可由dp[i-1][j-Ci] + Wi转移过来
b.不放,那么dp[i][j]的状态可由dp[i-1][j]转移过来
我们所要做的就是在这两种决策中选取一种最优的;
所以状态转移方程为dp[i][j] =max(dp[i-1][j-Ci] + Wi , dp[i-1][j]); - 按一个方向求出该问题的解
状态dp[i][j]总是要取决于之前的状态dp[i-1][j-Ci] +Wi,dp[i-1][j],所以我们应按按自底而上的方向求解
代码如下:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int INF = 2005;
int dp[INF][INF]; //定义状态dp[i][j]为前i件物品放入容量为j的背包时的最大价值;
int W[INF],C[INF];
int main()
{
int n, v;
scanf("%d%d",&n,&v);
for(int i = 1; i <= n; i++)
{
scanf("%d",&W[i]);
}
for(int i = 1; i <= n; i++)
{
scanf("%d",&C[i]);
}
memset(dp,0,sizeof(dp)); //初始化
for(int i = 1; i <= n; i++) //自底而上的求解
{
for(int j = 0; j <= v; j++)
{
if(j >= C[i])
{
dp[i][j] = max(dp[i-1][j],dp[i-1][j - C[i]] + W[i]); //状态转移
}
else
{
dp[i][j] = dp[i-1][j];
}
}
}
printf("%d\n",dp[n][v]);
return 0;
}