[优先队列] 洛谷 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

题解

• 这题的思想和上一题的思想差不多

代码

 1 #include<bits/stdc++.h>
 2 #define N 100010
 3 using namespace std;
 4 struct node{ int num,d,v;}k;
 5 priority_queue<node,vector<node>,less<node> >Q;
 6 bool operator <(node a,node b) {return a.v>b.v;}
 7 int n,m,a[N],b[N],c[N];
 8 int main()
 9 {
10     scanf("%d%d",&n,&m);
11     for (int i=1;i<=n;i++) scanf("%d%d%d",&a[i],&b[i],&c[i]),Q.push((node){i,1,a[i]+b[i]+c[i]});
12     while (m--)    printf("%d ",Q.top().v),k=Q.top(),Q.pop(),Q.push((node){k.num,k.d+1,a[k.num]*(k.d+1)*(k.d+1)+b[k.num]*(k.d+1)+c[k.num]});
13 }