代码随想录Day35：卡码网46.携带研究材料、416.分割等和子集

卡码网46. 携带研究材料

小明是一位科学家，他需要参加一场重要的国际科学大会，以展示自己的最新研究成果。他需要带一些研究材料，但是他的行李箱空间有限。这些研究材料包括实验设备、文献资料和实验样本等等，它们各自占据不同的空间，并且具有不同的价值。

小明的行李空间为 N，问小明应该如何抉择，才能携带最大价值的研究材料，每种研究材料只能选择一次，并且只有选与不选两种选择，不能进行切割。

题目链接：卡码网46.携带研究材料
文档讲解：代码随想录卡码网46.携带研究材料

题解

该题目为0-1背包问题，研究材料只能选择一次，所以只有带或不带两个选择。dp[i][j] 表示从下标为[0-i]的物品里任意取，放进容量为j的背包，价值总和最大是多少。

#include <iostream>
#include <vector>
using namespace std;
   
int main(){
    int M, N;
    cin >> M >> N;
    vector<int> weight(M, 0);
    vector<int> value(M, 0);
    for(int i = 0; i < M; i++) {
        cin >> weight[i];
    }
    for(int i = 0; i < M; i++) {
        cin >> value[i];
    }
    vector<vector<int>> dp(M, vector<int>(N + 1, 0));
    for(int j = weight[0]; j <= N + 1; j++) {
        dp[0][j] = value[0];
    }
    for(int i = 1; i < M; i++) {
        for(int j = 0; j <= N; j++) {
            if(weight[i] > j) dp[i][j] = dp[i - 1][j];
            else dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
        }
    }
    cout << dp[M - 1][N] << endl;
    return 0;
}

• 时间复杂度：O(MN)
• 空间复杂度：O(MN)

对于二维数组dp[M][N+1]，也可以只维护一维滚动数组dp[N+1]来实现。同时需要注意遍历背包容量的时候从大到小，避免重复添加已有的物品。

#include <iostream>
#include <vector>
using namespace std;
 
int main(){
    int M, N;
    cin >> M >> N;
    vector<int> weight(M, 0);
    vector<int> value(M, 0);
    for(int i = 0; i < M; i++) {
        cin >> weight[i];
    }
    for(int i = 0; i < M; i++) {
        cin >> value[i];
    }
    vector<int> dp(N + 1, 0);
    for(int i = 0; i < M; i++) {
        for(int j = N; j >= weight[i]; j--) {
            dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);
        }
    }
    cout << dp[N] << endl;
    return 0;
}

• 时间复杂度：O(MN)
• 空间复杂度：O(N)

---

416. 分割等和子集

给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。

题目链接：LeetCode416.分割等和子集
文档讲解：代码随想录LeetCode416.分割等和子集

题解

将题目转化为0-1背包问题，背包的容量就是sum(nums)/2，此时要求nums的和为偶数，数组中的数字就是要装的商品，重量和价值相等都是对应的值，如果dp数组中出现dp[sum(nums)/2]=sum(nums)/2说明可以分成等和子集。

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if (sum % 2 != 0)
            return false;
        int target = sum / 2;
        vector<int> dp(target + 1, 0);
        for (int i = 0; i < nums.size(); i++) {
            for (int j = target; j >= nums[i]; j--) {
                dp[j] = max(dp[j], dp[j - nums[i]] + nums[i]);
            }
        }
        if (dp[target] == target) {
            return true;
        }
        return false;
    }
};

• 时间复杂度：O(n^2)
• 空间复杂度：O(n)