洛谷2085 最小函数值

题目描述

有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个数应该输出到一行，用空格隔开。

输入输出样例

输入样例#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

本来先是用一个大数组，每次算出来三个值存进去进行堆排，发现这样有8个点T了，后来换了种思路，每次只计算出上一次最小函数的下一个值，再进行堆排

#include<bits/stdc++.h>
using namespace std;
const int MAXN=10005;
int n,m,f[MAXN],k[MAXN][4];
struct Alg{
    int num,data;
}a[MAXN];
bool cmp(Alg x,Alg y){
    return x.data<y.data;
} 
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d%d%d",&k[i][1],&k[i][2],&k[i][3]);
        a[i].num=i;
        a[i].data=k[i][1]+k[i][2]+k[i][3];
    }
    for(int i=1;i<=n;i++)
        f[i]=2;
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=m;i++){
        printf("%d ",a[1].data);
        int g=a[1].num;
        a[1].data=k[g][1]*f[g]*f[g]+k[g][2]*f[g]+k[g][3];
        f[g]++;
        int now=1;
        while(now*2<=n){
            int tp=2*now;
            if(a[tp].data>a[tp+1].data && tp<n) tp++;
            if(a[tp].data<a[now].data){
                Alg kx=a[tp];
                a[tp]=a[now];
                a[now]=kx; 
                now=tp;
            }
            else  break;
        }
    }
    return 0;
}