【洛谷】2095 营养膳食

Description

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

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

Input

第一行包含三个正整数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。

output

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

Analysis

按照脂肪指数排序，每次取最大的，证明略。

Code

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <cctype>
#include <cmath>
#include <time.h>
#include <map>
#include <set>
#include <vector>

using namespace std;

#define ms(a,b) memset(a,b,sizeof(a))

typedef long long ll;

struct rec{
    int a,b;
}q[210];
int n,m,k;
int r[110];

inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) {w|=ch=='-';ch=getchar();}
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}

bool cmp(rec x,rec y)
{
    return x.a>y.a;
}

int main()
{
    n=read(),m=read(),k=read();
    for (int i=1;i<=k;i++) r[i]=read();
    for (int i=1;i<=n;i++) q[i].a=read(),q[i].b=read();
    sort(q+1,q+1+n,cmp);
    int ans=0;
    for (int i=1;i<=n;i++)
    {
        if (m>0 && r[q[i].b]>0) ans+=q[i].a,m--,r[q[i].b]--;
    }
    printf("%d\n",ans);
    return 0;
}