【CJOJ2484】【Luogu2805】最小函数值（函数最小值）

题面

Description

有n个函数，分别为F1,F2,...,Fn。定义 Fi(x)=Aix2+Bix+Ci(x∈N∗)Fi(x)=Aix2+Bix+Ci(x∈N∗) 。给定这些Ai、Bi和Ci，请求出所有函数的所有函数值中最小的m个（如有重复的要输出多个）。

Input

第一行输入两个正整数n和m，n<=500000, m<=500000

以下n行每行三个正整数，其中第i行的三个数分别为Ai、Bi和Ci。输入数据保证Ai<=10，Bi<=100，Ci<=10000。

Output

输出将这n个函数所有可以生成的函数值排序后的前m个元素。

这m个数应该输出到一行，用空格隔开，并且最后一个数右侧也有一个空格。

Sample Input

3 10

4 5 3

3 4 5

1 7 1

Sample Output

9 12 12 19 25 29 31 44 45 54

题解

每个函数的a,b,c都是正数

所以每个抛物线的对称轴都是负数

所以每条抛物线在x>=1时都是增函数

所以fi(1)一定是该抛物线最小值

所以首先把所有的fi(1)都放到优先队列里面

每次直接输出优先队列的top值

然后把top值所对应的函数的x+1的值放到队列里面(因为单调递增)

循环n次就是答案

优先队列可以自己手打堆，也可以直接使用priority_queue

使用STL代码会非常简洁

#include<iostream>

#include<cstdio>

#include<cstring>

#include<cstdlib>

#include<cmath>

#include<queue>

#include<algorithm>

using namespace std;

#define ll long long

#define MAX 500010

struct Node

{

	    ll val;

	    ll X;

	    ll from;

};

bool operator <(Node a,Node b)

{

	   return a.val>b.val;

}

priority_queue<Node> fun;

ll a[MAX],b[MAX],c[MAX];

ll n,m;

inline ll Function(ll n,ll x)

{

	   return a[n]*x*x+b[n]*x+c[n];

}

int main()

{

       scanf("%lld%lld",&n,&m);

       ll S,num=0,x=0;

       for(int i=1;i<=n;++i)

       {

       	      scanf("%lld%lld%lld",&a[i],&b[i],&c[i]);

       	      fun.push((Node){Function(i,1),1,i});

       }

       for(ll i=1;i<=m;++i)

       {

       	    Node Q=fun.top();

       	    cout<<Q.val<<' ';

       	    fun.pop();

       	    fun.push((Node){Function(Q.from,Q.X+1),Q.X+1,Q.from});

       }

	   return 0;

}

秒客网