【笔试练习题】背包问题

题目描述：有若干瓶饮料，每种饮料只有一瓶，每种饮料的打分不同，价格也不同。你的手里有买饮料的总金额，要求买到的饮料的打分总和最高。

输入：

第一行是一个正整数，总金额（不大于1000）；

第二行是n个正整数，代表饮料单价（n不大于100）；

第三行是n个正整数，代表饮料的打分。

输出：最高总分

解析：这是经典的背包问题。背包容量——总金额；物品重量——单价；物品价值——热度值。

#include <iostream>
#include <vector>
#include <string>
#include <sstream>
#include <algorithm>
using namespace std;
/*
测试案例：
1000
200 600 100 180 300 450
6 10 3 4 5 8
*/
template <typename T>
void print_vec(vector<T> v){
	for(int i  = 0; i < v.size(); i++){
		cout << v[i] << " ";
	}
	cout << endl;
}

void get_data(int &money, vector<int> &price, vector<int> &score){
	cin >> money; 
	cin.ignore(); // cin和string::getline合起用时,需要清缓存
	vector<string> str(2, "");
	for(int i = 0; i < 2; i++){
		// 已知读取两行的情况下,使用string::getline读取一行的数据
		getline(cin, str[i]);
	}
	stringstream ss1(str[0]);
	int a;
	while(ss1 >> a){
		price.push_back(a);
	}
	stringstream ss2(str[1]);
	while(ss2 >> a){
		score.push_back(a);
	}
}


int main(int argc, char const *argv[])
{
	int money; // 总金额
	vector<int> price;
	vector<int> score;
	get_data(money, price, score);
	int n = price.size(); // 共有n瓶饮料
	vector<vector<int> > v(n + 1, vector<int> (money + 1, 0)); // 建立表
	
	for(int i = 1; i <= n; ++i){
		for(int j = 1; j <= money; ++j){
			if(j - price[i-1] < 0) // 买不起
				v[i][j] = v[i - 1][j];
			else
				v[i][j] = max(v[i - 1][j], v[i - 1][j - price[i - 1]] + score[i - 1]);
				// 解释: v[i - 1][j]不买这个物品 VS v[i - 1][j - price[i - 1]] + score[i - 1]买这个物品(剩余空间的最高价值 + 这个物品的价值)
                // 如果改成 v[i][j - price[i - 1]] + score[i - 1]，就是物品总数无限
			cout << i << ":" << j << ":" << v[i][j] << endl;
		}
	}
	cout << v[n][money] << endl;
	return 0;
}

二、要求输出购买方案

需要用到另一个矩阵。

#include <iostream>
#include <vector>
#include <string>
#include <sstream>
#include <algorithm>
using namespace std;
/*
100
20 60 10 18 30 45
6 10 3 4 5 8
*/
// 打印数组
template <typename T>
void print_vec(vector<T> v){
	for(int i  = 0; i < v.size(); i++){
		cout << v[i] << " ";
	}
	cout << endl;
}

// 从终端接收数据
void get_data(int &money, vector<int> &price, vector<int> &score){
	cin >> money; 
	cin.ignore(); // cin和string::getline合起用时,需要清缓存
	vector<string> str(2, "");
	for(int i = 0; i < 2; i++){
		// 已知读取两行的情况下,使用string::getline读取一行的数据
		getline(cin, str[i]);
	}
	stringstream ss1(str[0]);
	int a;
	while(ss1 >> a){
		price.push_back(a);
	}
	stringstream ss2(str[1]);
	while(ss2 >> a){
		score.push_back(a);
	}
}

// 获得放入的序列
void get_drink(vector<vector<bool> > &g, vector<int> &ans, int i, int j, vector<int> &price){
	if(i == 0)
		return ;
	if(g[i][j]){
		ans.push_back(i);
		get_drink(g, ans, i - 1, j - price[i - 1], price);
	}
	else {
		get_drink(g, ans, i - 1, j, price);
	}
}

int main(int argc, char const *argv[])
{
	int money; // 总金额
	vector<int> price;
	vector<int> score;
	get_data(money, price, score);
	// 以下为背包法代码
	int n = price.size(); // 共有n瓶饮料
	vector<vector<int> > v(n + 1, vector<int> (money + 1, 0)); // 建立表
	vector<vector<bool> > g(n + 1, vector<bool> (money + 1, false)); // 建立是否放入表
	for(int i = 1; i <= n; ++i){
		for(int j = 1; j <= money; ++j){
			if(j - price[i-1] < 0) // 买不起
				{
					v[i][j] = v[i - 1][j];
					g[i][j] = false;
				}
			else
				{
					int unchoose = v[i - 1][j];
					int choose = v[i - 1][j - price[i - 1]] + score[i - 1];
					// 解释: v[i - 1][j]不买这个物品 VS v[i - 1][j - price[i - 1]] + score[i - 1]买这个物品(剩余空间的最价值 + 这个物品的价值)
					if(unchoose > choose){
						v[i][j] = unchoose;
						g[i][j] = false;
					}
					else{
						v[i][j] = choose;
						g[i][j] = true;
					}
				}
		}
	}
	// 以上为背包法代码
	cout << v[n][money] << endl;
	vector<int> ans;
	get_drink(g, ans, n, money, price);
	print_vec(ans);
	return 0;
}