洛谷 P2085 最小函数值(minval)

最新推荐文章于 2024-09-17 14:19:28 发布

原创最新推荐文章于 2024-09-17 14:19:28 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #堆 #结构体

洛谷专栏收录该内容

184 篇文章

订阅专栏

本文介绍了一种使用结构体和优先队列求解多个二次函数在自然数范围内最小值的方法。通过不断更新每个函数的下一个可能最小值并进行比较，确保找到最小的m个函数值。

题目描述

有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个数应该输出到一行，用空格隔开。

输入输出样例

输入样例#1：

输出样例#1：

9 12 12 19 25 29 31 44 45 54

说明

数据规模：n,m<=10000

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

又刷新了对结构体的认知啊……实在是妙……

主要的思路很简单，就是结构体比较麻烦吧，还有重载运算符~

#include<cstdio>
#include<queue>
using namespace std;

struct node{
	int x,val,y;
	bool operator < (const node &c) const
	{
		return val>c.val;
	}
};

int n,m,a[3][10005],x;
priority_queue<node> q;

int kk(int u,int v)
{
	return a[0][u]*v*v+a[1][u]*v+a[2][u];
}

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d%d",&a[0][i],&a[1][i],&a[2][i]);
		q.push((node){i,kk(i,1),1});
	}
	while(m)
	{
		m--;
		node z=q.top();
		q.pop();
		printf("%d ",z.val);
		q.push((node){z.x,kk(z.x,z.y+1),z.y+1});
	}
	printf("\n");
	return 0;
}