蓝桥杯(C++ 最大开支 优先队列)

优先队列: 蓝桥杯(C++ 整数删除 优先队列 )-CSDN博客

蓝桥杯(C++ 最大开支 优先队列)_第1张图片

蓝桥杯(C++ 最大开支 优先队列)_第2张图片

思路: 

1、每个人依此选择项目,每个人选项目时都(选择当下花费增加最多的项目),若项目i的门票价格为kx+b,那么增加一个人选择的花费为:increase =(k*(x+1)+b)*(x+1)- ((k*x)+b)*x x = k(2x+1) + b

2、可以用优先队列存项目的k、b、x(选择这个项目的人数)、increase(当前项目人数增加为x+1所增加的花费),使increase最大的项目排在队首,每加一个人出队一个并使总花费增加increase,即总花费money = money + increase

3、更新x和increase,若increase > 0重新入队,若increase <= 0不重新入队(人数到了开口向下的二次函数的对称轴,花费开始下降,不能再增加人数了

代码:

#include
#include
#include
using namespace std;
using ll = long long;
struct node
{
    int k, b, x , increase;
};
bool operator < (const struct node a, const struct node b)//重载<
{
    return a.increase < b.increase;
}
int main()
{
    int n, m;
    cin >> n >> m;
    priority_queueh;//大根堆
    node a;
    for (int i = 0; i < m; i++)
    {
        cin >> a.k >> a.b;
        a.increase = a.k + a.b;
        if (a.increase <= 0)//小于等于零的不入队
            continue;
        a.x = 1;
        h.push(a);
    }
    ll money = 0, person;
    for (person = 0; !h.empty() && person < n; person++)
    {
        if (h.top().increase > 0)//防止第一个就小于零
        {
            node temp = h.top();
            h.pop();
            money += temp.increase;//加上增加的花费
            //increase=(k*(x+1)+b)*(x+1)-((k*x)+b)*x 再增加一个人会不会比之前花费更多
            temp.increase = temp.k * (2 * temp.x + 1) + temp.b;
            if (temp.increase > 0)//比之前还更多,重新入队
            {
                temp.x += 1;
                h.push(temp);
            }
        }
        else
            break;
    }
    cout << money;
}

 

你可能感兴趣的:(蓝桥杯,c++,开发语言,算法)