P2095 营养膳食

题目描述

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

为了增肥，Mr.L 希望吃到更多的脂肪。然而也不能只吃高脂肪食品，那样的话就会导致缺少其他营养。

Mr.L 通过研究发现：真正的营养膳食规定某类食品不宜一次性吃超过若干份。比如就一顿饭来说，肉类不宜吃超过 1 份，鱼类不宜吃超过 1 份，蛋类不宜吃超过 1 份，蔬菜类不宜吃过 2 份。

Mr.L 想要在营养膳食的情况下吃到更多的脂肪，当然 Mr.L 的食量也是有限的。

输入格式

第一行包含三个正整数 n,m 和 k。表示 Mr.L 每顿饭最多可以吃 m 份食品，同时有 n 种食品供 Mr.L 选择，而这 n 种食品分为 k 类。

第二行包含 k 个不超过 10 的正整数，表示可以吃 1 到 k 类食品的最大份数。

接下来 n 行每行包括 2 个正整数，分别表示该食品的脂肪指数 ai​ 和所属的类别 bi​。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

60

说明/提示

对于 100% 的数据，1≤n≤200，1≤m≤100，1≤k≤100，1≤ai​≤100，1≤bi​≤k。

代码：

#include<iostream> 
#include <algorithm>
using namespace std;
int x[105];
struct nod{//定义结构体存储n种食品的脂肪数ai和所属类别bi
	int a;
	int b;
}node[205];
bool cmp(nod x,nod y){//定义cmp函数规定结构体的排序规则
	if(x.a!=y.a) return x.a>y.a;
	else if(x.b!=y.b) return x.b<y.b;
}
int main(){
	
	int n,m,k;
	cin>>n>>m>>k;
	for(int i=1;i<=k;i++){
		cin>>x[i];
	}
	for(int i=1;i<=n;i++){
		cin>>node[i].a>>node[i].b;
	}
	sort(node+1,node+1+n,cmp);//对结构体进行排序
	int cnt=0;
	int sum=0;
	for(int j=1;j<=n;j++){
		if(x[node[j].b]>0&&sum<m){//遍历n种食品，选择的食品的类别不能超过xi,份数不能超过m
			cnt+=node[j].a;
			sum++;
			x[node[j].b]--;
		}	
	}
	cout<<cnt;
	return 0;
}