代码随想录算法训练营第四十一天| 卡码网第46题、416. 分割等和子集

卡码网第46题

题目链接
文章讲解

#include<bits/stdc++.h>
using namespace std;
 
int M, N;
 
void slove() {
    vector<int> weight(M);
    vector<int> value(M);
    for (int i = 0; i < M; i++) cin >> weight[i];
    for (int i = 0; i < M; i++) cin >> value[i];
    // 定义dp数组，dp数组初始化
    vector<int> dp(N + 1, 0);
    // 遍历推导dp数组
    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] << '\n';
}
 
int main() {
    while (cin >> M >> N) slove();
    return 0;
}

416. 分割等和子集

题目链接
文章讲解

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