一、背包问题
题目一:46. 携带研究材料
46. 携带研究材料(第六期模拟笔试)
时间限制:5.000S 空间限制:128MB
题目描述
小明是一位科学家,他需要参加一场重要的国际科学大会,以展示自己的最新研究成果。他需要带一些研究材料,但是他的行李箱空间有限。这些研究材料包括实验设备、文献资料和实验样本等等,它们各自占据不同的空间,并且具有不同的价值。
小明的行李空间为 N,问小明应该如何抉择,才能携带最大价值的研究材料,每种研究材料只能选择一次,并且只有选与不选两种选择,不能进行切割。
输入描述
第一行包含两个正整数,第一个整数 M 代表研究材料的种类,第二个正整数 N,代表小明的行李空间。
第二行包含 M 个正整数,代表每种研究材料的所占空间。
第三行包含 M 个正整数,代表每种研究材料的价值。
输出描述
输出一个整数,代表小明能够携带的研究材料的最大价值。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int knapsack(int M, int N, vector<int>& space, vector<int>& value) {
vector<vector<int>> dp(M + 1, vector<int>(N + 1, 0));
for (int i = 1; i <= M; ++i) {
for (int j = 1; j <= N; ++j) {
if (space[i - 1] > j) {
dp[i][j] = dp[i - 1][j];
} else {
dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - space[i - 1]] + value[i - 1]);
}
}
}
return dp[M][N];
}
int main() {
int M, N;
cin >> M >> N;
vector<int> space(M);
vector<int> value(M);
for (int i = 0; i < M; ++i) {
cin >> space[i];
}
for (int i = 0; i < M; ++i) {
cin >> value[i];
}
int result = knapsack(M, N, space, value);
cout << result << endl;
return 0;
}
题目二:46. 携带研究材料(滚动数组)
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int knapsack(int M, int N, vector<int>& space, vector<int>& value) {
vector<int> dp(N + 1, 0);
for (int i = 1; i <= M; ++i) {
for (int j = N; j >= space[i - 1]; --j) {
dp[j] = max(dp[j], dp[j - space[i - 1]] + value[i - 1]);
}
}
return dp[N];
}
int main() {
int M, N;
cin >> M >> N;
vector<int> space(M);
vector<int> value(M);
for (int i = 0; i < M; ++i) {
cin >> space[i];
}
for (int i = 0; i < M; ++i) {
cin >> value[i];
}
int result = knapsack(M, N, space, value);
cout << result << endl;
return 0;
}
二、分割等和子集
题目一:416. 分割等和子集
给你一个 只包含正整数 的 非空 数组 nums
。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
/*
* @lc app=leetcode.cn id=416 lang=cpp
*
* [416] 分割等和子集
*/
// @lc code=start
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;
int n = nums.size();
vector<vector<bool>> dp(n + 1, vector<bool>(target + 1, false));
for (int i = 0; i <= n; ++i) {
dp[i][0] = true;
}
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= target; ++j) {
if (nums[i - 1] <= j) {
dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]];
} else {
dp[i][j] = dp[i - 1][j];
}
}
}
return dp[n][target];
}
};
// @lc code=end