题目描述
有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个数应该输出到一行,用空格隔开。
写了几道堆排,感觉顺手多了。。。这道题题目中有一个描述是a,b,c都是正整数。根据初中数学之二次函数的顶点,则函数的最小值的x肯定是在负半轴(我就是一开始没想到),然后在定义域(0,+∞)上单调递增。
其实因为这道题数据很水(应该是的),然后暴力就能过,建一个大根堆,枚举每一个函数的1~m的值,如果满足条件就入堆(有没有这个名词。。)然后如果已经堆中的数大于等于了要求数,而且这个数大于堆顶那么就跳过这个函数。就这样。
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAXN=10005;
int n,m;
//int a[MAXN],b[MAXN],c[MAXN],cnt[MAXN];
int t;
int hp[MAXN];
int ans[MAXN];
struct O {
int a,b,c;
} h[MAXN];
int del() { //大根堆
int res=hp[1];
hp[1]=hp[t];
t--;
int now=1;
while(now*2<=t) {
int tp=now*2;
if(tp<t&&hp[tp]<hp[tp+1])tp++;
if(hp[tp]>hp[now])swap(hp[tp],hp[now]);
else break;
now=tp;
}
return res;
}
bool cmp(O x,O y) {
return x.c<y.c;
}
void pus(int x) {
hp[++t]=x;
int now=t;
while(now>1) {
int tp=now/2;
if(hp[now]>hp[tp])swap(hp[now],hp[tp]);
else break;
now=tp;
}
}
int f(int i,int x) {
return h[i].a*x*x+h[i].b*x+h[i].c;
}
int main() {
scanf("%d%d",&n,&m);
memset(hp,0x7f,sizeof hp);
for(register int i=1; i<=n; i++) {
scanf("%d%d%d",&h[i].a,&h[i].b,&h[i].c);
}
// sort(h+1,h+n,cmp);
for(register int i=1; i<=n; i++) {
for(register int j=1; j<=m; j++) {
int r=f(i,j);
while(t>m) {//堆内元素数量不大于m
del();
}
if(r<hp[1]||t<m) {
pus(r);
} else break;
}
}
while(t>m) {
del();
}
for(int i=1; i<=m; i++) {
ans[i]=del();
}
for(int i=m; i>=1; i--)cout<<ans[i]<<" ";//倒序输出
}
马上就12月月考了。。。square