有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
1 #include<iostream>
2 #include<algorithm>
3 #include<cstdio>
4 #include<queue>
5 using namespace std;
6 const int N = 1000000 + 5;
7 int a[N],b[N],c[N],p[N],n,m;
8 struct node
9 {
10 int a,b;
11 }f[N];
12 int function(int i,int x)
13 {
14 return a[i]*x*x+b[i]*x+c[i];
15 }
16 void update(int i)
17 {
18 if(f[i].a>f[i*2].a && i*2<=n)
19 {
20 swap(f[i],f[i*2]);
21 update(i*2);update(i/2);
22 }
23 if(f[i].a>f[i*2+1].a && i*2+1<=n)
24 {
25 swap(f[i],f[i*2+1]);
26 update(i*2+1);update(i/2);
27 }
28 }
29 int main()
30 {
31 scanf("%d%d",&n,&m);
32 for(int i=1;i<=n;i++)
33 {
34 p[i]=1;
35 scanf("%d%d%d",&a[i],&b[i],&c[i]);
36 f[i].a=function(i,p[i]++);f[i].b=i;
37 }
38 for(int i=1;i<=n;i++)update(i);
39 for(int i=0;i<m;i++)
40 {
41 printf("%d ",f[1].a);
42 f[1].a=function(f[1].b,p[f[1].b]++);
43 update(1);
44 }
45 return 0;
46 }