【一本通 1370】【堆】最小函数值(minval)

一本通 1370：最小函数值(minval)

---

题目

【题目描述】
有n个函数，分别为F1,F2,…,Fn。定义Fi(x)=Aix2+Bix+Ci(x∈N∗)。给定这些Ai、Bi和Ci，请求出所有函数的所有函数值中最小的m个（如有重复的要输出多个）。

【输入】
第一行输入两个正整数n和m。
以下n行每行三个正整数，其中第i行的三个数分别位Ai、Bi和Ci。输入数据保证Ai<=10，Bi<=100，Ci<=10000。

【输出】
将这n个函数所有可以生成的函数值排序后的前m个元素。这m个数应该输出到一行，用空格隔开。

【输入样例】

3 10
4 5 3
3 4 5
1 7 1

【输出样例】

9 12 12 19 25 29 31 44 45 54

【数据规模】
n,m≤10000。

---

解题思路

首先最能想到的方法是用小根堆维护，但是发现MLE还RE
于是开始思考怎样把堆中的数维护在 m 个数内
如果是for循环判断T到你怀疑人生
此时转变思路，每次维护 m 个数时肯定是用最大的数和当前数比，最大的数可以用大根堆找出
于是摒弃小根堆而用大根堆维护，最后倒着输出就好了

---

Code

#include <bits/stdc++.h>
#define ll long long 
#define Inf 2147483647 
#define N 10000

using namespace std;

int n, m, a, b, c, mid, now, ls[N + 200];
priority_queue<int> q;

int main() {
	scanf("%d %d", &n, &m);
	for(int i = 1; i <= m; i ++) q.push(Inf);
	for(int i = 1; i <= n; i ++) {
		scanf("%d %d %d", &a, &b, &c);
		mid = -b / (2 * a);  //对称轴
		mid = max(1, mid);   //避免负数
		for(int j = max(mid - m + 1, 1), k = 0; k <= m; j ++, k ++) {
			now = a * j * j + b * j + c;
			if(q.size() < m || now < q.top()) {  //与最大数比较
				q.pop();
				q.push(now);
			} else break;  //如果当前数已经更新不了了，那么后面的数一定更大
		}
	}
	for(int i = 1; i <= m; i ++) {
		ls[i] = q.top();
		q.pop();
	}
	for(int i = m; i; i --) printf("%d ", ls[i]);  //倒着输出
}