洛谷 堆专题 最小函数值

题目描述

有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

题解：因为a,b,c>0所以fi(x)在(0,+∞)上是增函数，且x是自然数。

我们建立一个大根堆存储当前最小的M个函数值，对于每一个fi(x)我们只需求出1-m的函数值与堆顶比较，如果比堆顶小就置换堆顶，如果比堆顶大就直接break（fi(x)是增函数所以后面的函数值一定比前面的大）。最后按倒序输出堆中元素即可

#include <iostream>
#include <cstdio>
#include <queue>
#define MAXN 10000
using namespace std;
priority_queue <int> heap;
int n,m;
int ans[MAXN+5];

int calc(int a,int b,int c,int x)
{
	return a*x*x+b*x+c;
}

int main()
{
	int a,b,c;
	scanf("%d%d",&n,&m);
	scanf("%d%d%d",&a,&b,&c);
	for (int i=1;i<=m;i++) heap.push(calc(a,b,c,i)); 
	for (int i=2;i<=n;i++)
   	{
   		scanf("%d%d%d",&a,&b,&c);
   		for (int j=1;j<=m;j++)
   		{
   			if (calc(a,b,c,j)<heap.top())
			{
				heap.pop();
				heap.push(calc(a,b,c,j));
			}
			else break; 
   			
   		}
   	}
   	for (int i=1;i<=m;i++) 
   	{
   		ans[i]=heap.top();
   		heap.pop();
   	}
   	for (int i=m;i>=1;i--)
   	{
   		printf("%d ",ans[i]);
   	}
   	return 0;
}