2018拼多多校招笔试贪心编程题小熊吃糖详解

题目描述

有n只小熊，他们有着各不相同的战斗力。每次他们吃糖时，会按照战斗力来排，战斗力高的小熊拥有优先选择权。前面的小熊吃饱了，后面的小熊才能吃。每只小熊有一个饥饿值，每次进食的时候，小熊们会选择最大的能填饱自己当前饥饿值的那颗糖来吃，可能吃完没饱会重复上述过程，但不会选择吃撑。

现在给出n只小熊的战斗力和饥饿值，并且给出m颗糖能填饱的饥饿值。

求所有小熊进食完之后，每只小熊剩余的饥饿值。

输入描述:

第一行两个正整数n和m，分别表示小熊数量和糖的数量。（n <= 10, m <= 100）
第二行m个正整数，每个表示着颗糖能填充的饥饿值。
接下来的n行，每行2个正整数，分别代表每只小熊的战斗力和当前饥饿值。
题目中所有输入的数值小于等于100。

输出描述:

输出n行，每行一个整数，代表每只小熊剩余的饥饿值。

示例1

输入

2 5
5 6 10 20 30
4 34
3 35

输出

4
0

说明

第一只小熊吃了第5颗糖
第二只小熊吃了第4颗糖
第二只小熊吃了第3颗糖
第二只小熊吃了第1颗糖

这是一道简单的贪心题目，每次战斗力最大的熊吃能量最高的食物即可，用一个数组记录食物是否被吃。要特别注意，最后输出的时候需要原来的顺序，所以要将原来的顺序记录下来

这里多次使用到自定义排序算法，推荐C++11中的lamda表达式。

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 200;
typedef pair<int, int> PII;
int main()
{
	int n, m;
	int candy[N];
	bool eaten[N] = { false };       // 标记糖果是否被吃

	PII bear[N];
	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		cin >> candy[i];
	}
	for (int i = 0; i < n; i++)
		cin >> bear[i].first >> bear[i].second;

	sort(candy, candy + m, greater<int>());
	sort(bear, bear + n, [](const PII& a, const PII& b) {
		return a.first > b.first; });

	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			if (candy[j] <= bear[i].second && !eaten[j]) {
				bear[i].second -= candy[j];
				eaten[j] = true;
			}
		}
	}

	for (int i = 0; i < n; i++) {
		cout << bear[i].second << endl;
	}
}