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

此题大体思路和序列合并： https://blog.youkuaiyun.com/SunPeishuai/article/details/80254580类似，不再注释；

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=10002;
priority_queue<int,vector<int> >h;//大根堆
int a[maxn],b[maxn],c[maxn];
int n,m,now;
int main()
{
  scanf("%d %d",&n,&m);
  for(int i=1;i<=n;i++)
    scanf("%d %d %d",&a[i],&b[i],&c[i]);
  for(int i=1;i<=m;i++)
    h.push(a[1]*i*i+b[1]*i+c[1]);
  for(int i=2;i<=n;i++)
  for(int j=1;j<=m;j++)
  {
      now=a[i]*j*j+b[i]*j+c[i];
     if(now<h.top())
     {
         h.pop();
         h.push(now);

     }
     else break;
  }
  for(int i=1;i<=m;i++)
  {
      c[i]=h.top();
      h.pop();
  }
  for(int i=m;i>=1;i--)
    printf("%d ",c[i]);
  return 0;
}