Папизи
Профи
(878)
9 часов назад
#include <iostream>
#include <vector>
#include <algorithm>
int main() {
int n, W;
std::cin >> n >> W;
std::vector<int> weights(n);
std::vector<int> values(n);
for (int i = 0; i < n; ++i) {
std::cin >> weights[i];
}
for (int i = 0; i < n; ++i) {
std::cin >> values[i];
}
// Интернетирующий массив для хранения максимальной стоимости
std::vector<int> dp(W + 1, 0);
// Массив для отслеживания выбранных предметов
std::vector<std::vector<bool>> selected(n, std::vector<bool>(W + 1, false));
// Основной процесс наполнения рюкзака
for (int i = 0; i < n; ++i) {
for (int w = W; w >= weights[i]; --w) {
if (dp[w] < dp[w - weights[i]] + values[i]) {
dp[w] = dp[w - weights[i]] + values[i];
selected[i][w] = true;
}
}
}
// Определяем максимальную стоимость
int max_value = dp[W];
// Чтобы вывести набор предметов, которые составляют этот максимальный набор
std::vector<int> items;
for (int i = n - 1, w = W; i >= 0; --i) {
if (selected[i][w]) {
items.push_back(i + 1); // Сохраняем номера предметов (индексация с 1)
w -= weights[i]; // уменьшаем оставшуюся ёмкость
}
}
// Получение количества предметов
std::cout << items.size() << std::endl;
// Вывод номера предметов в порядке убывания
std::reverse(items.begin(), items.end());
for (int item : items) {
std::cout << item << " ";
}
std::cout << std::endl;
return 0;
}
-
Ученик
(77)
9 часов назад
# Данные
weights = [2, 4, 1, 2]
values = [7, 2, 5, 1]
max_weight = 4
# Результат
max_value = knapsack(weights, values, max_weight)
print(max_value)
Выполнение кода
Если мы запустим этот код:
Для предмета 1 с массой 2 и стоимостью 7: максимальная стоимость при весе 2 - 7.
Для предмета 2 с массой 4 и стоимостью 2: максимальная стоимость при весе 4 - 7 (берём предмет 1).
Для предмета 3 с массой 1 и стоимостью 5: можем взять его с предметом 1 (вес 3, стоимость 12) или без него (максимальное получается при весе 4 - 7).
Для предмета 4 с массой 2 и стоимостью 1: вес 2 уже занят, но стоимость не изменится - максимальная всё равно 7.
Результат
Максимальная стоимость, которую можно унести в рюкзаке с весом не более 4, составляет 12 (предметы 1 и 3: масса 2 + 1 = 3, стоимость 7 + 5 = 12).
Входные данные
4 6
2 4 1 2
7 2 5 1
Результат работы
3
4 3 1