题目链接
题意:
有n个二次函数,每个函数中x有一个正整数值,这n个x的值和为m,求这n个函数计算结果的和的最小值。
思路:
因为这n个x都是正整数,所以让每个函数中x的初始值都为1,然后记录每个函数的x增加1时,这个函数值增加的值,把这些值丢到一个优先队列中,每次不断更新序列,直到进行m-n次 也就是直到所有x的和为m为止。
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn = 100005;
struct hh
{
ll a, b, c, x;
ll num;
bool operator<(const hh &a) const
{
return num > a.num;
}
} k[maxn];
priority_queue<hh> que;
int main()
{
ll n, m;
while (~scanf("%lld%lld", &n, &m))
{
ll i, j, ans = 0;
for (i = 1; i <= n; i++)
{
scanf("%lld%lld%lld", &k[i].a, &k[i].b, &k[i].c);
k[i].x = 1;
k[i].num = k[i].a * ((k[i].x + 1) * (k[i].x + 1) - (k[i].x) * (k[i].x)) + k[i].b;
que.push(k[i]);
}
for (i = 1; i <= m - n; i++)
{
hh k1 = que.top();
que.pop();
k1.x++;
//printf("***%lld\n",k1.a);
k1.num = k1.a * ((k1.x + 1) * (k1.x + 1) - (k1.x) * (k1.x)) + k1.b;
que.push(k1);
}
for (i = 1; i <= n; i++)
{
hh k2 = que.top();
ans += k2.a * k2.x * k2.x + k2.b * k2.x + k2.c;
que.pop();
}
printf("%lld\n", ans);
}
}