洛谷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;
}