P2085 最小函数值——小顶堆、贪心、重载运算符

题目描述

有 \(n\) 个函数，分别为 \(F_1,F_2,\dots,F_n\)。定义 \(F_i(x)=A_ix^2+B_ix+C_i(x\in\mathbb N*)\)。给定这些 \(A_i\)、\(B_i\) 和 \(C_i\)，请求出所有函数的所有函数值中最小的 \(m\) 个（如有重复的要输出多个）。

输入格式

第一行输入两个正整数 \(n\) 和 \(m\)。

以下 \(n\) 行每行三个正整数，其中第 \(i\) 行的三个数分别为 \(A_i\)、\(B_i\) 和 \(C_i\)。

输出格式

输出将这 \(n\) 个函数所有可以生成的函数值排序后的前 \(m\) 个元素。这 \(m\) 个数应该输出到一行，用空格隔开。

输入输出样例 #1

输入 #1

3 10
4 5 3
3 4 5
1 7 1

输出 #1

9 12 12 19 25 29 31 44 45 54

说明/提示

数据规模与约定

对于全部的测试点，保证 \(1 \leq n,m\le10000\)，\(1 \leq A_i\le10,B_i\le100,C_i\le10^4\)。

题解

#include <iostream>
#include <queue>
#include <vector>
using namespace std;// 定义结构体存储函数值、函数编号和当前 x 值
struct Node {int value;  // 函数值int funcIndex;  // 函数编号int x;  // 当前 x 值// 重载运算符，用于优先队列的比较bool operator>(const Node& other) const {return value > other.value;}
};int main() {int n, m;cin >> n >> m;// 存储每个函数的 A, B, C 参数vector<vector<int>> functions(n, vector<int>(3));for (int i = 0; i < n; ++i) {cin >> functions[i][0] >> functions[i][1] >> functions[i][2];}// 定义小根堆priority_queue<Node, vector<Node>, greater<Node>> heap;// 初始化小根堆for (int i = 0; i < n; ++i) {int A = functions[i][0];int B = functions[i][1];int C = functions[i][2];// 计算 x = 1 时的函数值int value = A * 1 * 1 + B * 1 + C;// 将 (函数值, 函数编号, 当前 x 值) 加入堆heap.push({value, i, 1});}// 存储结果vector<int> result;// 进行 m 次操作for (int i = 0; i < m; ++i) {// 取出堆中最小的函数值Node top = heap.top();heap.pop();result.push_back(top.value);// 计算下一个 x 值对应的函数值int A = functions[top.funcIndex][0];int B = functions[top.funcIndex][1];int C = functions[top.funcIndex][2];int nextX = top.x + 1;int nextValue = A * nextX * nextX + B * nextX + C;// 将新的函数值加入堆heap.push({nextValue, top.funcIndex, nextX});}// 输出结果for (int i = 0; i < m; ++i) {if (i > 0) {cout << " ";}cout << result[i];}cout << endl;return 0;
}

贪心策略