AcWing 01背包问题 题解

dp专题、经典题目

01背包问题

题目

思路

每个物品有拿与不拿两种状态（拿的时候还需要判断一下能不能装得下）。考虑第i个物品时，前i-1个物品的最大价值，相对于前i个物品的最大价值，属于重叠子问题，这样就可以想到前i个物品的最大价值由前i-1个物品的最大价值得来：

• 当不拿第i个物品时，f[i] = f[i-1]
• 当能拿第i个物品时，就需要考虑，单单一个状态没法表示所有的子集状态：前多少物品、体积多少的最大价值；这时就要多加一个状态f[i][j] = f[i-1][j - v[i]] + w[i]；表示前i-1个物品且体积最多为j-v[i]时的最大价值加上第i个物品的价值w[i]
• 毕竟两种情况，取大的值就是前i个物品的最大价值

代码

记忆递归

由于0<N,V≤1000，故递归不会爆栈，这个数据规模可以AC

#include <bits/stdc++.h>
using namespace std;

const int N = 1010;
int f[N][N], v[N], w[N];

int dfs(int i, int j){
    if(f[i][j] != -1) return f[i][j];
    if(i == 0) return 0;
    int res = dfs(i-1, j);
    if(v[i] <= j) res = max(res, dfs(i-1, j - v[i]) + w[i]);
    return f[i][j] = res;
}

int main(){
    int n, m;
    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] = -1;

    cout<< dfs(n, m) << endl;
    return 0;
}

dp

假如数据规模更大，会爆栈时，就需要将记忆递归修改为循环实现

#include <bits/stdc++.h>
using namespace std;

const int N = 1010;
int f[N][N], v[N], w[N];

int main(){
    int n, m;
    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(v[i] <= j) f[i][j] = max(f[i][j], f[i-1][j-v[i]] + w[i]);
        }
    cout<< f[n][m] << endl;
    return 0;
}

dp-滚动数组

每一行的状态只与前一行的状态有关，所以可以用两个一维数组交替记录就可以实现。细想一下，每一行的状态只与前一行的当前位置向前有关，如果从后往前更新，就不会覆盖掉前一个状态，所以一个一维数组就可以实现。

#include <bits/stdc++.h>
using namespace std;

const int N = 1010;
int f[N], v[N], w[N];

int main(){
    int n, m;
    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 --)
            f[j] = max(f[j], f[j - v[i]] + w[i]);

    cout<< f[m] << endl;
    return 0;
}

总结

01背包题目常见，毕竟容易想到。y总说了

dp就得多练