codeforces 496e Distributing Parts 贪心

最新推荐文章于 2020-07-06 19:29:11 发布

AC_vitasoy

最新推荐文章于 2020-07-06 19:29:11 发布

阅读量833

点赞数

本文链接：https://blog.csdn.net/u011580493/article/details/42059053

版权

OJ——Codeforces 同时被 2 个专栏收录

49 篇文章 0 订阅

订阅专栏

基本算法——贪心

8 篇文章 0 订阅

订阅专栏

题意：

现在有n个曲子，每个曲子的范围为ai~bi。有m个演奏家，每个演奏家的范围为ci~di，并且可以出演次数为ki次。

如果ci<= ai<=bi<=di，则说明该曲子可以由演奏家演出。

让你找出合理的方案使得所有曲子都能被演奏，无方案输出“NO”;

思路：

这题跟区域赛上海站比较像，虽说当时挂0了。。。。

首先根据r值，也就是右边的值，把两个数组从小到大排个序。（降维，把r值的限制去掉）

开始遍历演奏家，把节目的bi值，小于等于演奏家的di值的节目，全扔到set里，set里根据ai排序。（看到铭神用stl用得灰起，有空我也得研究一下set - -）。

然后二分lower_bound找出与演奏家的ci值最相近的节目，说明该演奏家会演奏该节目。如果无合适的节目可选，跳到下一个演奏家。

code：

#include <set>
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;

typedef long long LL;
const int MAXN = 1e5+5;

struct PP
{
    int l, r, k;
    int rank, ans;
    bool operator < (const PP& b) const
    {
        if(r == b.r) return l < b.l;
        return r < b.r;
    }
}a[MAXN], b[MAXN];

struct Point
{
    int l, r, rank;
    bool operator < (const Point& b) const 
    {
        if(l == b.l)
        {
            if(r == b.r) return rank < b.rank;
            return r < b.r;
        }
        return l < b.l;
    }
};

int n, m;
int res[MAXN];

void input()
{
    cin>>n;
    for(int i = 0;i < n; i++)
    {
        cin>>a[i].l>>a[i].r;
        a[i].rank = i;
    }
    cin>>m;
    for(int i = 0;i < m; i++)
    {
        cin>>b[i].l>>b[i].r>>b[i].k;
        b[i].rank = i;
    }
}
bool cmp(const PP &a, const PP &b)
{
    return a.rank < b.rank;
}
void solve()
{
    sort(a, a+n);
    sort(b, b+m);
    set <Point> s;
    int cnt = n;
    for(int i = 0, j = 0;i < m; i++)
    {
        while(j < n && a[j].r <= b[i].r)
        {
            s.insert((Point){a[j].l, a[j].r, j});
            j++;
        }
        for(int z = 0; z < b[i].k; z++)
        {
            auto it = s.lower_bound((Point){b[i].l, 0, 0});
            if(it == s.end()) break;
            a[it->rank].ans = b[i].rank+1;
            s.erase(it);
            cnt--;
        }
    }

    //output answer
    if(s.size() != 0 || cnt != 0)
    {
        cout<<"NO"<<endl;
        return;
    }
    sort(a, a+n, cmp);
    cout<<"YES"<<endl;
    cout<<a[0].ans;
    for(int i = 1; i < n; i++)
        cout<<" "<<a[i].ans;
    cout<<endl;
}

int main()
{
    ios::sync_with_stdio(false);
    input();
    solve();
    return 0;
}