【DP】01背包问题

一、题目

2. 01背包问题 - AcWing题库高质量的算法题库https://www.acwing.com/problem/content/2/

二、题解

基本思路

状态表示

dp[i][j]表示前i个物品中选择不超过j体积的最大价值

状态转移方程

第i个物品分为选和不选两种情况, dp[i][j]在两种情况中选择最大值即可‘

如果不选第i个物品，那么dp[i][j] 其实就 从i-1个物品中选择体积不超过j的最大价值

如果选择第i个物品，那么dp[i][j]就是w[i] + dp[i-1][j - v[i]]，就是第i个物品的价值加上在第i-1个物品中选择j减去第i个物品重量的最大价值，但需要注意如果选择第i个位置的物品，要求是此时的j必须大于等于该物品的重量

dp[i][j] = max(dp[i - 1][j], w[i] + dp[i - 1][j - v[i]];

初始化

在读入数据时下标从数组1开始，填dp时从左到右从上到下，dp[0][j]中的值默认全部为0

返回值

由状态定义可知，只需要返回前n个物品中选择体积不超过V的最大价值也即是dp[n][V]

#include<iostream>
#include<cstring>
using namespace std;
const int N = 1010;
int n, V, v[N], w[N], dp[N][N];

int main()
{
    cin >> n >> V;
    for (int i = 1; i <= n; i++) cin >> v[i] >> w[i];
    
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= V; j++)
        {
            dp[i][j] = dp[i - 1][j];
            if (j >= v[i])
                dp[i][j] = max(dp[i][j], dp[i - 1][j - v[i]] + w[i]);
        }
    
    cout << dp[n][V] << endl;
    return 0;
}

滚动数组优化

对于上述二维数组dp可以通过滚动数组进行优化，将其优化到只需要两个数组，本题可以之间优化到一个数组，在上述代码进行填表过程中不难发现在填第i行时只需要i - 1行数据即可，我们可以维护两个数组，不断进行滚动即可完成上述操作，但是在填第i， j时，只需要i-1,j或者i-1, j-v[i]， 这个时候我们可能之间在一个数组上从右往左填表即可完成上述两个数组的操作

#include<iostream>
#include<cstring>
using namespace std;
const int N = 1010;
int n, V, v[N], w[N], dp[N];

int main()
{
    cin >> n >> V;
    for (int i = 1; i <= n; i++) cin >> v[i] >> w[i];
    
    for (int i = 1; i <= n; i++)
        for (int j = V; j >= v[i]; j--)
            dp[j] = max(dp[j], dp[j - v[i]] + w[i]);
    
    cout << dp[V] << endl;
    return 0;
}