题目描述
有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个数应该输出到一行,用空格隔开。
输入输出样例
说明
数据规模:n,m<=10000
此题大体思路和序列合并:
https://blog.csdn.net/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; }