AcWing 2. 01背包问题

Spring.A

已于 2022-02-14 15:03:01 修改

阅读量264

点赞数 1

分类专栏：动态规划文章标签： c++ 动态规划算法

于 2022-02-14 15:01:20 首次发布

本文链接：https://blog.csdn.net/weixin_60484917/article/details/122924384

版权

动态规划专栏收录该内容

24 篇文章 0 订阅

订阅专栏

参考题目：
AcWing 2. 01背包问题

状态表示：

f[i][j] ：在前 i 个物品中选，且总体积不超过 j 的所有方案。答案为 f[n][m]
属性 ：总价值的 Max

状态计算：

选 i 的：f[i - 1][j - v[i]] + w[i]
不选 i 的：f[i - 1][j]
综上 f[i][j] = max(f[i - 1][j], f[i - 1][j - v[i]] + w[i])

代码：

朴素版

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1010;

int n, m;
int v[N], w[N];
int f[N][N];

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

优化成一维的

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1010;

int n, m;
int v[N], w[N];
int f[N];

int main()
{
    cin >> n >> m;
    
    for (int i = 1; i <= n; i ++ ) cin >> v[i] >> w[i];
    
    for (int i = 1; i <= n; i ++ )
        for (int j = m; j >= v[i]; j -- ) // 01背包需要逆序循环
            f[j] = max(f[j], f[j - v[i]] + w[i]);
    
    cout << f[m] << endl;
    
    return 0;
}