luogu P2085 最小函数值

题目传送门:https://www.luogu.org/problemnew/show/P2085



题意:

有n个函数F1,F2,F3,……,FnFi(x)=Ai*x^2+Bi*x+Ci (x∈n)),求其中最小的m个。



思路:

开一个小根堆,因为这个x是我们自己枚举的,所以开一个f[i]表示对于第i个函数,我们的x取到了多少。

然后我们可以将x=1的所有情况推进堆中,每次取出堆顶,因为堆顶有且仅属于一个函数,那么我们对应的f[i]++即可(因为x的值可以取更大)。

这种一出一近的操作使得堆内始终保持n个元素,所以时间复杂度为O(mlogn)。



代码:

#include<cstdio>
#include<queue>
using namespace std;
struct node
{
	int id,date;
	friend bool operator<(const node &x,const node &y)
	{
		return x.date>y.date;
	}
};
priority_queue<node>a;
using namespace std;
	int p[20000][5],f[20000];
	int n,m;
int Function(int x,int y)
{
	return p[x][1]*y*y+p[x][2]*y+p[x][3];
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d %d %d",&p[i][1],&p[i][2],&p[i][3]);
		a.push((node){i,Function(i,1)});
		f[i]=1;
	}
	for(int i=1;i<=m;i++)
	{
		node now=a.top();
		a.pop();
		printf("%d ",now.date);
		a.push((node){now.id,Function(now.id,++f[now.id])});
	}
}

阅读更多
换一批

没有更多推荐了,返回首页