1370：最小函数值

【题目描述】

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

【题目链接】

　　　　http://ybt.ssoier.cn:8088/problem_show.php?pid=1370

【算法】

　　　　大根堆记录，一旦总个数超过m个则删去堆顶。

【代码】

#include <bits/stdc++.h>
using namespace std;
int n,m,i,a,b,c,j;
int heap[10010],tot;
int calc(int a,int b,int c,int x) { return a*x*x+b*x+c; }
void up(int p)
{
    while(p>1)
        if(heap[p]>heap[p/2]) swap(heap[p],heap[p/2]),p/=2;
        else break;
}
void Insert(int val)
{
    heap[++tot]=val;
    up(tot);
}
void down(int p)
{
    int s=p*2;
    while(s<=tot) {
        if(s<tot&&heap[s+1]>heap[s]) s++;
        if(heap[s]>heap[p]) swap(heap[s],heap[p]),p=s,s*=2;
        else break;
    }
}
void Extract()
{
    heap[1]=heap[tot--];
    down(1);
}
void print()
{
    int cur=heap[1];
    if(tot>1) Extract(),print(),printf(" %d",cur);
    else if(tot==1) printf("%d",cur);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++) {
        scanf("%d%d%d",&a,&b,&c);
        for(j=1;j<=m;j++) {
            int cur=calc(a,b,c,j);
            if(tot<m) Insert(cur);
            else if(cur>=heap[1]) break;
            else Extract(),Insert(cur);
        }
    }
    print();
    return 0;
}