洛谷P2085——最小函数值

题目描述

有n个函数，分别为(F_1,F_2,...,F_n)。定义(F_i(x)=A_i*x^2+B_i*x+C_i (x∈N*))。给定这些(A_i、B_i和C_i)，请求出所有函数的所有函数值中最小的m个（如有重复的要输出多个）。

输入格式

输入数据：第一行输入两个正整数n和m。以下n行每行三个正整数，其中第i行的三个数分别位Ai、Bi和Ci。(Ai<=10，Bi<=100，Ci<=10 000)。

输出格式

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

思路

堆排序，注意要加greater<int>，详见此篇文章

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
priority_queue <ll, vector<ll>, greater<ll> > heap;
int main() {
    register ll n,m;
    scanf("%lld%lld",&n,&m);
    for(register int i = 0;i<n;++i) {
        register ll a,b,c;
        scanf("%lld%lld%lld",&a,&b,&c);
        for(register int j = 1;j<=100;++j) {
            heap.push((int)a*j*j+b*j+c);
        }
    }
    while(m--) {
        printf("%lld ",heap.top());
        heap.pop();
    }
    return 0;
}