【51nod 3241】【堆】小明和他的同学们

51nod 3241 小明和他的同学们

---

题目

小明在全球数学竞赛获得了冠军，他很高兴，所以他想请他的同学吃巧克力，但是小明每天都很忙，只有x分钟的时间可以宴请同学们，现在小明有m个巧克力，小明要请n个同学吃巧克力，每个同学吃一块巧克力的时间都不一样，有的同学吃的快，有的同学吃的慢，每个同学吃完一块巧克力之后，马上就会去吃另外一个巧克力，假如2个人同时吃完，优先发给吃的快的同学。直到没有剩余的巧克力或者宴请时间结束了。

现在小明想要知道在宴请结束之后有几块巧克力被吃完，有几块巧克力没有被吃完，但是被吃了一些，聪明的你可以帮助小明解决这个问题吗？

注意，如果两个人同时吃完当前巧克力，那么吃的快的同学优先得到巧克力。

输入
第一行包括一个t，表示测试样例的个数；
对于每个测试样例：
第一行包括三个整数n，m，x，表示有n个同学，m个巧克力，x分钟的宴请时间；
第二行包括n个整数，每个整数表示该同学吃一块巧克力的时间。
输出
每个测试样例输出一行，每行包括两个整数 a，b，表示a块巧克力被完整吃完，b块巧克力被吃了但是没有被完整吃完。
数据范围
对于10%的数据，1<= n<= 4；
对于50%的数据，1<= n<= 100；
对于100%的数据，1<= n<=100；
1<=m<=5000，1<=x<=1000，1<=t<=20，1<=每个同学吃一块巧克力的时间<=1000。
输入样例

3
1 2 1
1
3 8 5
1 3 4
5 4 1
5 4 3 2 1

输出样例

1 0
7 1
1 3

---

解题思路

给谁吃巧克力很好解决，只需要用一个小根堆记录吃巧克力速度，每次取堆顶吃，然后将堆顶弹出
❓但是怎么记录谁吃完了，把他重新丢进堆里呢
把吃巧克力的人丢进另一个小跟堆里，记录的是他什么时候吃完巧克力，不断判断堆顶吃完没
如果吃完了，就把他弹出，丢进等巧克力的堆里
❓怎么记录吃完的巧克力数和吃一半的巧克力数
当丢进吃巧克力堆里时，吃一半巧克力数加一
当吃巧克力堆弹出时，吃一半巧克力数减一，吃完数加一

❗选谁吃巧克力是一个单独分开的工作
因为选谁不算时间，但是又要保证有下一秒让他吃
类似“下一秒的准备工作”，比如最开始没吃时，就要选谁在第一秒吃，这就是第一秒的准备工作，类似0.5秒做

---

Code

#include <bits/stdc++.h>

using namespace std;

int T, n, m, x, ansx, ansy, a[200];
priority_queue<pair<int, int> > p, q;

int main() {
	scanf("%d", &T);
	while(T --) {
		ansx = ansy = 0;
		while(!q.empty()) q.pop();
		while(!p.empty()) p.pop();
		scanf("%d %d %d", &n, &m, &x);
		for(int i = 1; i <= n; i ++) {
			scanf("%d", &a[i]);
			q.push(make_pair(-a[i], i));  //丢进等巧克力列中
		}
		if(x > 0)
			for(int j = 1, k = min(n, m); j <= k; j ++, m --) {  //选第一秒谁吃
				p.push(make_pair(q.top().first, q.top().second));   //p是吃巧克力堆
				q.pop();
				ansy ++;
			}
		for(int i = 1; i <= x; i ++) {  //枚举宴会时间
			while(!p.empty() && -p.top().first == i) {  //判断第i秒有没有人吃完
				ansy --, ansx ++;  //吃一半巧克力数减一，吃完数加一
				int x = p.top().second;
				p.pop(), q.push(make_pair(-a[x], x));
			}
			for(int j = 1, k = min(n, m); j <= k && x - i > 0 && !q.empty(); j ++, m --) {
				p.push(make_pair(-(i + (-q.top().first)), q.top().second));
				q.pop();
				ansy ++;
			}
		}
		printf("%d %d\n", ansx, ansy);
	}
}