解题思路
所以我们建一个大根堆,存最小的数到第m小的数,第m小的数就理所当然的是堆顶了。 每次我们只需要比较新加进来的数比堆顶大还是比堆顶小。
如果比堆顶小,将原来的堆顶丢掉,将新的数塞进去; 如若比堆顶大,根据该题题意,a>0&&b>0,函数对称轴
x
=
−
b
2
∗
a
x=-\frac{b}{2*a}
x=−2∗ab/恒小于0,可以得出,y在
x
>
0
x>0
x>0时是单调递增的,所以接下来的函数值y只会大不会小,可以直接
b
r
e
a
k
break
break掉了
由于我们存储的时候用的是大根堆,所以记得要逆序输出,将m个数从小到大输出嗷
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<iomanip>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#define ll long long
using namespace std;
int n,m,a,b,c,ans[10010];
priority_queue<int> q;
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&a,&b,&c);
for(int j=1;j<=m;j++)
{
int k=a*j*j+b*j+c;
if(i==1)
q.push(k);
else
{
if(k<q.top())
{
q.pop();
q.push(k);
}
else break;
}
}
}
for(int i=1;i<=m;i++)
{
ans[i]=q.top();
q.pop();
}
for(int i=m;i>0;i--)
printf("%d ",ans[i]);
}