01背包问题

背包问题老是做了忘忘了做，索性这次分析记录下来

一个可装载重量为W的背包和N个物品，每个物品有重量和价值两个属性。其中第i个物品的重量为wt[i]，价值为val[i]，现在让你用这个背包装物品，最多能装的价值是多少？
例： N = 3, W = 4 wt = [2, 1, 3] val = [4, 2, 3]

Output：6

明确状态：背包容量，可选择物品

定义dp[i][j]： dp[i][j]表述，对于前i个物品，当前背包的容量为j，这种情况下可以装的最大价值是dp[i][j]

Base Case: dp[i][0]=dp[0][i]=0

选择：装入背包或者不装入背包,那么dp[i][j]=max( dp[i-1][j] , dp[i-1][j-w[i-1]]+val[i-1])
代码

int knapsack(int W, int N, vector<int>& wt, vector<int>& val) {
    // vector 全填入 0，base case 已初始化
    vector<vector<int>> dp(N + 1, vector<int>(W + 1, 0));
    for (int i = 1; i <= N; i++) {
        for (int w = 1; w <= W; w++) {
            if (w - wt[i-1] < 0) {
                // 当前背包容量装不下，只能选择不装入背包
                dp[i][w] = dp[i - 1][w];
            } else {
                // 装入或者不装入背包，择优
                dp[i][w] = max(dp[i - 1][w - wt[i-1]] + val[i-1], 
                               dp[i - 1][w]);
            }
        }
    }

    return dp[N][W];
}

参考文章：东哥的算法小抄