题目描述
有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
出题人善意地把二次函数的对称轴都放在了y轴左侧,所以所有的函数都是单调递增,维护一个小根堆,即可。
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
int n,m,a[10001],b[10001],c[10001];
struct data
{
int val,knd,x;
}w[10001],tmp;
bool operator < (const data &c,const data &d)
{
return c.val>d.val;
}
priority_queue<data>q;
int f(int i,int x)
{
return a[i]*x*x+b[i]*x+c[i];
}
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<=n;i++)
{
tmp.val=f(i,1);
tmp.knd=i;
tmp.x=1;
q.push(tmp);
}
while(m--)
{
tmp=q.top();
q.pop();
printf("%d ",tmp.val);
tmp.x++;
tmp.val=f(tmp.knd,tmp.x);
q.push(tmp);
}
printf("\n");
return 0;
}