当前位置: 代码迷 >> 综合 >> CodeForces - 496E Distributing Parts (贪心+数据结构应用)
  详细解决方案

CodeForces - 496E Distributing Parts (贪心+数据结构应用)

热度:106   发布时间:2023-11-15 14:38:56.0

题目链接: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;
}

 

  相关解决方案