P2085 最小函数值

题目描述

有n个函数，分别为F1,F2,...,Fn。定义Fi(x)=Ai*x^2+Bi*x+Ci (x∈N*)。给定这些Ai、Bi和Ci，请求出所有函数的所有函数值中最小的m个（如有重复的要输出多个）。

输入输出格式

输入格式：

 

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

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

3 10
4 5 3
3 4 5
1 7 1

输出样例#1： 复制

9 12 12 19 25 29 31 44 45 54

说明

数据规模：n,m<=10000

用一个优先队列，让队列的头为最大值，维护这个优先队列就好。

代码如下：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
#define ll long long
using namespace std;
struct ff{
   int ai;
   int bi;
   int ci;
}f[10005];

int F(int a,int b,int c,int x){
    return a*x*x+b*x+c;
}
int main(){
   int n,m;
   while(cin>>n>>m){
    ll cc[100005];
    memset(cc,0,sizeof(cc));
    int i,j;
    priority_queue<int ,vector<int>,less<int> > a;
    for(i=0;i<n;i++){
        cin>>f[i].ai>>f[i].bi>>f[i].ci;
    }

        for(j=1;j<=m;j++){
            a.push(F(f[0].ai,f[0].bi,f[0].ci,j));
        }
        for(i=1;i<n;i++){
            for(j=1;j<=m;j++){
                if(a.top()>F(f[i].ai,f[i].bi,f[i].ci,j)){
                      a.pop();
                      a.push(F(f[i].ai,f[i].bi,f[i].ci,j));
                }
                else break;
            }
        }
    for(i=0;i<m;i++){
        cc[i]=a.top();
        a.pop();
    }
    sort(cc,cc+m);
    for(i=0;i<m;i++){
        if(i==0){
            cout<<cc[i];
        }
        else{
            cout<<" "<<cc[i];
        }
    }
    cout<<endl;
   }
   return 0;
}