P2085 最小函数值(minval)

题目描述

有n个函数，分别为F1,F2,...,Fn。定义Fi(x)=Aix^2+Bix+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

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<queue>
using namespace std;
const int N = 1000000 + 5; 
int a[N],b[N],c[N],p[N],n,m; 
struct node
{
    int a,b;
}f[N];
int function(int i,int x)
{
    return a[i]*x*x+b[i]*x+c[i];
}
void update(int i)
{
    if(f[i].a>f[i*2].a && i*2<=n)
    {
        swap(f[i],f[i*2]);
        update(i*2);update(i/2);
    }
    if(f[i].a>f[i*2+1].a && i*2+1<=n)
    {
        swap(f[i],f[i*2+1]);
        update(i*2+1);update(i/2);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        p[i]=1;
        scanf("%d%d%d",&a[i],&b[i],&c[i]);
        f[i].a=function(i,p[i]++);f[i].b=i;
    }
    for(int i=1;i<=n;i++)update(i);
    for(int i=0;i<m;i++)
    {
        printf("%d ",f[1].a);
        f[1].a=function(f[1].b,p[f[1].b]++);
        update(1);
    }
    return 0;
}