CodeForces - 496E Distributing Parts （贪心+数据结构应用）

本文详细解析了CodeForces平台上的E题，通过将人物和区间视为二元组并进行排序，使用set数据结构来解决区间覆盖问题，确保每个人物的贡献仅在其所属区间内体现，最终实现最优分配。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接:http://codeforces.com/problemset/problem/496/E

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

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
const int  maxn =1e5+5;
const int mod=1e9+7;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}

/*
题目大意：给定n个part,m个人，
每个人都至多可以有k个part,一个人拥有一个part的条件就是，
区间包含（端点重复也可以）。

有点想之前做过的树状数组玩线段的题目哇。。。
首先在模拟的时候我们可以发现，我们把人和二元组都看成二元组进行排序，
那么我们需要保证，在扫倒一个人的时候，这个人的贡献只在之前扫到的part部分出现，
很容易想到是按右端点排序，再按左端点从大到小排(模拟一下就行)，
再按种类排序（同理），只要保证人的贡献完全在左边。

那么在左边维护出一个part的有序集合，容易想到set或者map，
那么对于扫倒的人，我们把左端点丢到set中去lowerbound找出
上确界，问题就转化为数个有序端点，一个人只能取有限次，
从哪端开始选最优，明显从左往右扫就最优，保证以后出现的人能得到更多的东西。
整体思路就出来了。

*/

struct node
{
    int l,r,op,id;///结束时间
    bool operator<(const node& y)const
    {
        if(r==y.r)
        {
            if(l==y.l) return op<y.op;///如果边界都一样part放在前面
            return l>y.l;///结束时间一样
        }
        return r<y.r;
    }
}p[maxn<<1];

int n,m,v[maxn];
struct bin
{
  int l,id;
  bool operator<(const bin& y) const
  {
      if(l==y.l) return id<y.id;
      return l<y.l;
  }
};

set<bin> s;
set<bin>::iterator it,tp;

int ans[maxn];

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&p[i].l,&p[i].r);
        p[i].id=i,p[i].op=0;///不是人而是part
    }
     scanf("%d",&m);
     for(int i=n+1;i<=n+m;i++)
     {
         scanf("%d%d%d",&p[i].l,&p[i].r,&v[i-n]);
         p[i].id=i-n,p[i].op=1;
     }
     sort(p+1,p+1+n+m);///把整体按结束时间从小到大排序

     int cnt=0;
     for(int i=1;i<=n+m;i++)
     {
        if(p[i].op)
        {
            it=s.lower_bound(bin{p[i].l,0});
            if(it==s.end()) continue;///不能选择
            while(it!=s.end()&&v[p[i].id]--)
            {
                ans[it->id]=p[i].id;
                cnt++;
                s.erase(it++);
            }
        }
        else s.insert(bin{p[i].l,p[i].id});
     }
     if(cnt==n)
     {
         puts("YES");
        for(int i=1;i<=n;i++) printf("%d%c",ans[i],i==n?'\n':' ');
     }
     else puts("NO");
    return 0;
}