P2085 最小函数值

题目描述

有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

与前面的一题类似，不多赘述

#include<bits/stdc++.h>

using namespace std;

struct f
{
    int quan;
    int node;
    int a;
    int b;
    int c;
    f (int u , int v, int w , int t , int q)
    {
        a = u;
        b = v;
        c = w;
        quan = t;
        node = q;
    }
};

bool operator<(const f &a , const f &b)
{
    return a.quan > b.quan;
}

priority_queue<f>que;

int main()
{
    int n , m ;
    int u ,v , w;
    cin >> n >> m;
    for(int i = 1; i <= n ; i ++)
    {
        scanf("%d %d %d" , &u , &v , &w);
        que.push(f( u , v , w , u + v + w , 1));
    }
    int num = 0;
    while(num < m)
    {
        f temp = que.top();
        que.pop();
        num ++;
        printf("%d" , temp.quan);
        if(num != m)
            printf(" ");
        int t = temp.node + 1;
        que.push(f(temp.a , temp.b ,temp.c , temp.a * t * t + temp.b * t + temp.c ,t ));
    }
    return 0;
}