背包问题

题目链接

0/1背包

暴力解法

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

int main(){
    int N, V;
    cin >> N >> V;
    int v[N], w[N];
    vector<vector<int> > dp(N+1, vector<int>(V+1, 0));
    for(int i = 0; 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-1] >= 0)
                dp[i][j] = max(dp[i-1][j], dp[i-1][j - v[i-1]] + w[i-1]);
        }

    }

    cout << dp[N][V];
    return 0;
}

优化空间复杂度

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

int main(){
    int N, V;
    cin >> N >> V;
    int v[N], w[N];
    vector<int> dp(V+1, 0);
    for(int i = 0; i < N; i++)
        cin >> v[i] >> w[i];

    for(int i = 1; i <= N; i++){
        for(int j = V; j >= 1; j--){  // 注意这里的遍历方向问题（反向）
            if(j - v[i-1] >= 0)
                dp[j] = max(dp[j], dp[j - v[i-1]] + w[i-1]);
        }

    }

    cout << dp[V];
    return 0;
}

完全背包

暴力解法

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

int main(){
    int N, V;
    cin >> N >> V;
    int v[N], w[N];
    vector<vector<int> > dp(N+1, vector<int>(V+1, 0));
    for(int i = 0; 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-1] >= 0)
                dp[i][j] = max(dp[i-1][j], dp[i][j - v[i-1]] + w[i-1]);
        }

    }

    cout << dp[N][V];
    return 0;
}

优化空间复杂度

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

int main(){
    int N, V;
    cin >> N >> V;
    int v[N], w[N];
    vector<int> dp(V+1, 0);
    for(int i = 0; i < N; i++)
        cin >> v[i] >> w[i];

    for(int i = 1; i <= N; i++){
        for(int j = 1; j <= V; j++){  // 注意这里的遍历方向问题（正向）
            if(j - v[i-1] >= 0)
                dp[j] = max(dp[j], dp[j - v[i-1]] + w[i-1]);
        }

    }

    cout << dp[V];
    return 0;
}