这道题暴力解法就是把所有函数的前m个值代入算出来,然后把每个前m值的数组都合并起来,前m个就是我们的结果,当然这种做法是会超时的
所以我们应该选择 优先级队列,我们代入1把所有的值加入优先级队列,每次输出最小值,并且把该序列的第二个数代入进去加进队列,直到输出m个数结束
#include <iostream>
#include <queue>
using namespace std;
const int N = 1e5+10;
typedef long long ll;
ll a[N],b[N],c[N];
ll n,m;
ll calc(ll pos,ll num)
{return a[pos]*num*num+b[pos]*num+c[pos];
}
struct node{ll sum;//函数值 ll pos;//第i个函数 ll num;//代入值 bool operator< (const node& x) const{return sum > x.sum;}
};
priority_queue<node> heap;
int main()
{cin >> n >> m;for(int i = 1;i<=n;i++){cin >> a[i] >> b[i] >> c[i];heap.push({calc(i,1),i,1});}for(int i = 1;i<=m;i++){node t = heap.top();heap.pop();ll sum = t.sum;ll num = t.num; ll pos = t.pos;cout << sum << " ";heap.push({calc(pos,num+1),pos,num+1});}return 0;
}