https://www.luogu.org/recordnew/show/17149258

题解:

C++版本一

 #include <iostream>
    using namespace std;
    int main()
    {
        int n,m,i,j,cmin,jmin;
        int A[10010], B[10010], C[10010];
        int F[10010];
        cin>>n>>m;
        for(i=0;i<n;i++)
        {
            cin>>A[i]>>B[i]>>C[i];
            F[i]=1;
        }
        for(i=0;i<m;i++)
        {
            cmin=100000000;
            for(j=0;j<n;j++)
            {
                if(A[j]*F[j]*F[j]+B[j]*F[j]+C[j]<cmin)
                {
                    cmin=A[j]*F[j]*F[j]+B[j]*F[j]+C[j];
                    jmin=j;
                }
            }
            cout<<A[jmin]*F[jmin]*F[jmin]+B[jmin]*F[jmin]+C[jmin]<<' ';
            F[jmin]++;
        }
        return 0;
    }

C++版本二

小根堆

/*
*@Author:   STZG
*@Language: C++
*/
#include <bits/stdc++.h>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<bitset>
#include<queue>
#include<deque>
#include<stack>
#include<cmath>
#include<list>
#include<map>
#include<set>
//#define DEBUG
#define RI register int
using namespace std;
typedef long long ll;
//typedef __int128 lll;
const int N=100000+10;
const int M=100000+10;
const int MOD=1e9+7;
const double PI = acos(-1.0);
const double EXP = 1E-8;
const int INF = 0x3f3f3f3f;
int t,n,m,k,q;
int ans,cnt,flag,temp,sum;
char str;
struct node{
    int a,b,c,f;//每个函数的"箭头"位置;
}e[N];
struct DUI{
    int val;//箭头表示的函数值
    int x;//每个函数都有被输入进来的先后顺序,这个是第x个输入进来的函数
    //因为堆里面的节点总是在变化的,所以我们要记录哪个函数在哪个位置
}a[10010];
int heap_size;//堆的大小
void CHANGE(int m, int n){//自己写的交换函数
    DUI t;
    t=a[m];
    a[m]=a[n];
    a[n]=t;
}
void MIN_HEAPIFY(int i){
    int l=i*2;//右子节点
    int r=i*2+1;//左子节点
    int smallest;//记录父子节点值最小的那个
    if(l<=heap_size&&a[l].val<a[i].val)
        smallest=l;
    else
        smallest=i;
    if(r<=heap_size&&a[r].val<a[smallest].val)
        smallest=r;//父子节点中值最小的位置
    if(smallest!=i)//父节点最大则不变
    {
        CHANGE(i,smallest);//子节点大则交换父子节点
        MIN_HEAPIFY(smallest);//交换后继续往下维护
    }
}
void BUILD_HEAP(){//建立小根堆
    int i;
    for(i=heap_size/2;i>0;i--)
        MIN_HEAPIFY(i);//自底向上建堆
}
int F(int x){
    return e[x].a*e[x].f*e[x].f+e[x].b*e[x].f+e[x].c;
}
int main()
{
#ifdef DEBUG
	freopen("input.in", "r", stdin);
	//freopen("output.out", "w", stdout);
#endif
    //ios::sync_with_stdio(false);
    //cin.tie(0);
    //cout.tie(0);
    //scanf("%d",&t);
    //while(t--){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>e[i].a>>e[i].b>>e[i].c;
        e[i].f=1;
        a[i].val=F(i);
        a[i].x=i;//输入的顺序,第i个被输进来的
    }
    heap_size=n;
    BUILD_HEAP();
    for(int i=0;i<m;i++){
        cout<<a[1].val<<' ';//输出最小函数值
        e[a[1].x].f++;//它所在的函数中的"箭头"往后移
        a[1].val=F(a[1].x);//"箭头"变则值变
        MIN_HEAPIFY(1);//自顶向下维护堆
    }
    //}

#ifdef DEBUG
	printf("Time cost : %lf s\n",(double)clock()/CLOCKS_PER_SEC);
#endif
    //cout << "Hello world!" << endl;
    return 0;
}