问题 H: 深入浅出学算法050-营养膳食

p.s.自用

题目描述

Mr.L正在完成自己的增肥计划。

为了增肥，Mr.L希望吃到更多的脂肪。然而也不能只吃高脂肪食品，那样的话就会导致缺少其他营养。Mr.L通过研究发现：真正的营养膳食规定某类食品不宜一次性吃超过若干份。比如就一顿饭来说，肉类不宜吃超过1份，鱼类不宜吃超过1份，蛋类不宜吃超过1份，蔬菜类不宜吃超过2份。Mr.L想要在营养膳食的情况下吃到更多的脂肪，当然Mr.L的食量也是有限的。

输入

第一行包含三个正整数n（n≤200）,m（m≤100）和k（k≤100）。表示Mr.L每顿饭最多可以吃m份食品，同时有n种食品供Mr.L选择，而这n种食品分为k类。第二行包含k个不超过10的正整数，表示可以吃1到k类食品的最大份数。接下来n行每行包括2个正整数，分别表示该食品的脂肪指数ai和所属的类别bi，其中ai≤100，bi≤k。

输出

包括一个数字即Mr.L可以吃到的最大脂肪指数和。

样例输入 Copy

6 6 3
3 3 2
15 1
15 2
10 2
15 2
10 2
5 3

样例输出 Copy

60

思路

只考虑对ai进行从大到小的排序，然后在条件内选取尽可能大的ai。

参考答案

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <string>
#include <iomanip>

using namespace std;

int main()
{
	// 输入
	// 第一行
	int n, m, k; // n种食品可分为k类，有m份食品
	cin >> n >> m >> k;
	// 第二行
	vector<int> num(k + 1, 0); // 种类
	for (int i = 1; i <= k; i++)
		cin >> num[i];
	// 第3-n+3行
	vector<int> ai(n, 0); // 脂肪指数
	vector<int> bi(n, 0); // 所属的类别
	for (int i = 0; i < n; i++)
		cin >> ai[i] >> bi[i];

	// 求解
	// 冒泡排序：根据ai从大到小排
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n - i - 1; j++)
		{
			if (ai[j] < ai[j + 1])
			{
				int tempa = ai[j];
				ai[j] = ai[j + 1];
				ai[j + 1] = tempa;
				int tempb = bi[j];
				bi[j] = bi[j + 1];
				bi[j + 1] = tempb;
			}
		}
	}
	// 计算
	int res = 0; // 脂肪指数和
	int all_num = 0; // 吃了多少份食物
	for (int i = 0; i < n && all_num < m; i++)
	{
		int index = bi[i];
		if (num[index]) // 非空
		{
			res += ai[i];
			num[index] -= 1;
			all_num += 1;
		}
	}

	// 输出
	cout << res << endl;
}