题目描述
有n个函数,分别为F1,F2,…,Fn。定义Fi(x)=Aix^2+Bix+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
【解题思路】:
首先对于每个函数,当x=1时肯定是最小的。所以我们建一个大根堆,存最小的数到第m小的数,第m小的数就理所当然的是堆顶了。 每次我们只需要比较新加进来的数比堆顶大还是比堆顶小,如果比堆顶小,将原来的堆顶丢掉,将新的数塞进去; 如若比堆顶,根据该题题意,a>0&&b>0,函数对称轴x=-b/2*ax=−b/2∗a恒小于0,可以得出,y在x>0时是单调递增的,所以接下来的函数值y只会大不会小,可以直接break掉了由于我们存储的时候用的是大根堆,所以记得要逆序输出,将m个数从小到大输出即可。
【AC代码】:
#include<bits/stdc++.h>
#define M(a,b) memset(a,b,sizeof(a))
#define pi 3.1415926
using namespace std;
priority_queue<int ,vector<int> >q;
int n,m,a[10005];
int i,j,x,y,z,k;
int main(){
scanf("%d %d",&n,&m);
for(i=1;i<=n;i++){
scanf("%d %d %d",&x,&y,&z);
for(j=1;j<=m;j++){
k=x*j*j+y*j+z;
if(i==1)q.push(k);
else{
if(k<q.top()){
q.push(k);
q.pop();
}
else break;
}
}
}
for(i=1;i<=m;i++){
a[i]=q.top();
q.pop();
}
for(i=m;i>=1;i--)printf("%d ",a[i]);
return 0;
}