Aiiage Camp Day1 H Littrain wanna be rich

题意

　　N个苹果，A和B对每个苹果有一个出价ai，bi。每个苹果只能被卖给一个人，A最多买A个苹果，B最多买B个。求最多能卖多少钱。

　　A+B<=N<=1e5

　　ai,bi<=1e5

题解

　　按照ai排序，先由A买苹果1..A个。1..A必然会被买（若有苹果未被买，则可以让A少买一个后面的苹果使总价值最大）。那么对于剩下的苹果，先由B选前B大的。枚举苹果A+1..B：若此苹果没人买，则可以让B放弃当前最小苹果；若已被B买了，则让B放弃此苹果，而买1..A中bi-ai最大的苹果。过程记录最大值即可。

#include <bits/stdc++.h>
using namespace std;

struct node
{
    int a, b;
}p[100010];

bool cmp(node x, node y)
{
    if (x.a == y.a)
        return x.b > y.b;
    else
        return x.a > y.a;
}

set<pair<int, int > > aa, bb;

int main()
{
    int n, A, B;
    scanf("%d%d%d", &n, &A, &B);
    for (int i = 1; i <= n; ++i)
    {
        scanf("%d", &p[i].a);
        if (p[i].a < 0)
            p[i].a = 0; 
    }
    for (int i = 1; i <= n; ++i)
    {
        scanf("%d", &p[i].b);
        if (p[i].b < 0)
            p[i].b = 0;
    }
    sort(p + 1, p + 1 + n, cmp);
    long long m(0), ans(0);
    for (int i = 1; i <= A; ++i)
    {
        aa.insert({p[i].b - p[i].a, i});
        m += p[i].a;
    }
    for (int i = A + 1; i <= A + B; ++i)
    {
        bb.insert({p[i].b, i});
        m += p[i].b;
    }
    for (int i = A + B + 1; i <= n; ++i)
    {
        m += p[i].b;
        bb.insert({p[i].b, i}); 
        m -= bb.begin()->first;
        bb.erase(bb.begin());
    }
    ans = m;
    for (int i = A + 1; i <= A + B; ++i)
    {
        if (bb.find({p[i].b, i}) == bb.end())
        {
            m -= bb.begin()->first;
            bb.erase(bb.begin());
        }
        else
        {
            m -= p[i].b;
            bb.erase({p[i].b, i});
        }
        m += p[i].a;
        aa.insert({p[i].b - p[i].a, i});
        m += (--aa.end())->first;
        aa.erase(--aa.end());
        ans = max(ans, m);
    }
    printf("%lld
", ans);
    
    return 0;
}