【题解】LuoGu2085:最小函数值

原题传送门
做法跟这道题目差不多

Code:

#include 
#define maxn 10010
using namespace std;
struct heap{
	int x, val;
	bool operator < (const heap &x) const{ return x.val < val; }
};
priority_queue <heap> q;
int n, m, a[maxn], b[maxn], c[maxn], pos[maxn];

inline int read(){
	int s = 0, w = 1;
	char c = getchar();
	for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
	for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
	return s * w;
}

int calc(int id, int x){ return a[id] * x * x + b[id] * x + c[id]; }

int main(){
	n = read(), m = read();
	for (int i = 1; i <= n; ++i){
		a[i] = read(), b[i] = read(), c[i] = read();
		q.push((heap){i, calc(i, pos[i] = 1)});
	}
	for (int i = 1; i <= m; ++i){
		heap tmp = q.top(); q.pop();
		printf("%d ", tmp.val);
		q.push((heap){tmp.x, calc(tmp.x, ++pos[tmp.x])});
	}
	return 0;
}

你可能感兴趣的:(题解,LuoGu,优先队列(堆))