E.Education （ multiset的二分查找 ）

E.Education （ multiset的二分查找 ）

题目链接：https://vjudge.net/problem/Gym-101606E

题意：有n批人，第i批有ai个人，有m个公寓，第i个公寓可以容纳pi人，租金是ri，要求从这m个公寓里选出n个给这n批人，且总租金最小，输出第i批人应该租的公寓编号。

思路：贪心的考虑，肯定优先选择最便宜的公寓，那么这个公寓给谁呢？应该给小于这个公寓人数且最接近它的（ 联想到upper_bound() - 1 ）。公寓好说，存结构体里按照租金排序。n批人呢？我们应该对这n批人的需求排序，再进行二分查找。但是考虑到每次选择完一个公寓就应该从二分列表里删除，但数组是不好做删除操作的。我们考虑用multiset.

 

 

代码：

#include <bits/stdc++.h>

using namespace std;

int n,m;
struct node {
    int weight,money,id;
}a[5005];
struct nop {
    int id,people;
    bool operator<( const nop&b )const {
        return people<b.people;
    }
}t;
int ans[5005];
multiset<nop> s;

int rule( node a, node b )
{
    return a.money<b.money;
}

int main()
{
    cin >> n >> m;
    for ( int i=0; i<n; i++ ) {
        scanf("%d",&t.people);
        t.id=i+1;
        s.insert(t);
    }
    for ( int i=0; i<m; i++ ) scanf("%d",&a[i].weight);
    for ( int i=0; i<m; i++ ) scanf("%d",&a[i].money);
    for ( int i=0; i<m; i++ ) a[i].id=i+1;
    sort(a,a+m,rule);
    for ( int i=0; i<m; i++ ) { // 从最便宜的公寓开始枚举
        int now = a[i].weight;
        t.people = now;
        if ( s.size()==0 ) break;
        multiset<nop>::iterator it = s.upper_bound(t); // 找到满足当前公寓+1的批次
        if ( it!=s.begin() ) it --; // --到满足当前公寓的批次
        if ( (*it).people<=now ) {
            ans[ (*it).id ] = a[i].id;
            s.erase(it);
        }
    }
    if ( s.size()==0 ) for ( int i=1; i<=n; i++ ) cout << ans[i] << " ";
    else cout << "impossible" << endl;

    return 0;
}