CodeForces - 1315D Recommendations (贪心+multiset)

题目链接

一开始的想法是让物品按数量从小到大或者从大到小贪心，然而怎么想都不对，最后发现应该按价格贪

首先求出最终的物品数量序列（扫一遍即可）

然后把所有物品按价格从大到小排序，每次选价格最大的去匹配尽量小的数量（必须大于等于原来的数量），用multiset实现即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10,mod=1e9+7,inf=0x3f3f3f3f;
int n,a[N];
struct D {int a,b;} d[N];
bool cmp1(const D& x,const D& y) {return x.a<y.a;}
bool cmp2(const D& x,const D& y) {return x.b>y.b;}
multiset<int> st;
ll solve() {
    ll ret=0;
    for(int i=1; i<=n; ++i)st.insert(a[i]);
    sort(d+1,d+1+n,cmp2);
    for(int i=1; i<=n; ++i) {
        auto x=st.lower_bound(d[i].a);
        ret+=(ll)(*x-d[i].a)*d[i].b;
        st.erase(x);
    }
    return ret;
}
int main() {
    scanf("%d",&n);
    for(int i=1; i<=n; ++i)scanf("%d",&d[i].a);
    for(int i=1; i<=n; ++i)scanf("%d",&d[i].b);
    sort(d+1,d+1+n,cmp1);
    for(int i=1; i<=n; ++i)a[i]=max(a[i-1]+1,d[i].a);
    printf("%lld\n",solve());
    return 0;
}