最小函数值
题目
解析
先考虑暴力,显然假如把一切可能值都插入,是O(nmlogn),T(nm)的,TLE+MLE
这时就可以利用一下从这一题中学到的方法:在弹出一个必定比它小的值前,一定不需要将它塞入堆中,这样就可以优化到O((m+n)logn),T(n)的复杂度,轻松AC
code:
#include<cstdio>
using namespace std;
inline bool idigit(char x){
return (x<'0'|x>'9')?0:1;}
inline int read()
{
int num=0,f=1;
char c=0;
while(!idigit(c=getchar())){
if(c=='-')f=-1;}
while(idigit(c))num=(num<<1)+(num<<3)+(c&15),c=getchar();
return num*f;
}
inline void write(int x)
{
int F[20];
int tmp=x>0?x:-x;
if(x<0)putchar('-');
int cnt=0;
while(tmp>0){
F[cnt++]=tmp%10+'0';tmp/=10;}
while(cnt>0)putchar(F[--cnt]);
if(x==0)putchar('0');
putchar(' ');
}
struct q
{
int id,val;
bool operator< (const q x){
return val<x.val;}
bool operator> (const q x){
return val>x.val;}
void operator= (const q x){
val=x.val,id=x.id;}
}o2;
inline void swap(q &x,q &y){
q z=x;x=y,y=z;}
struct heap
{
int tot;
q a[10010];
inline void clean(){
tot=0;}
inline void up(int x){
while((x>>1)&&(a[x]<a[x>>1]))swap(a[x],a[x>>1]),x>>=1;}
inline void down(int x){
int r;while((x<<1)<=tot){
r=((x<<1|1)<=tot&&a[x<<1|1]<a[x<<1])?(x<<1|1):(x<<1);if(a[x]>a[r])swap(a[x],a[r]),x=r;else break;}}
inline int size(){
return tot;}
inline q top(){
return a[1];}
inline void pop(){
swap(a[1],a[tot--]),down(1);}
inline void push(q x){
a[++tot]=x,up(tot);}
}a;
int n,m,x[10010],y[10010],z[10010],tot[10010];
inline int f(int xx,int yy){
return x[xx]*yy*yy+y[xx]*yy+z[xx];}
int main()
{
n=read(),m=read(),a.clean();
for(int i=1;i<=n;++i)x[i]=read(),y[i]=read(),z[i]=read(),tot[i]=1,o2.id=i,o2.val=f(i,1),a.push(o2);
while(m--)
{
o2=a.top(),a.pop(),write(o2.val);
o2.val=f(o2.id,++tot[o2.id]),a.push(o2);
}
return 0;
}