problem:
给你n首歌,和每首歌的音调区间[a, b], a<=b
给你m个歌唱家,和每个歌唱家的音调区间[c, d]他最多可以唱k首歌, c<=d
一个歌唱家可以唱一首歌 iff c<=a<=b<=d
问能不能给出一种方案,使得每首歌都有歌唱家唱,能的话输出YES和每首歌由哪个歌唱家唱[1, m];不能输出NO
1<=n, m<=1e5
1<=a, b, c, d, k<=1e9
think:
1.对于a,b,c,d可以离散化,这样最后有2*n+2*m种音调。防止爆long long,k超多n的变成n
2.离线,按照音调左区间排序,当处理第i首歌的时候,把所有的cj<=ai的歌唱家都已经放进我们维护的数据结构,这样就不用考虑左区间了,只找右区间大于当前歌曲的歌唱家。
3.于是我们要维护每个右区间位置的歌唱家的信息。
4.如第一个样例,有5个音调值 1, 2, 3, 4, 5,开始的时候每个音调都是0,
1 4 2 => (0, 0, 0, 2, 0)
1 3 => (0, 0, 0, 1, 0) //这一步找到>=3的最小的不是0的位置,是4
2 5 1 => (0, 0, 0, 1, 1)
2 4 => (0, 0, 0, 0, 1)
3 5 => (0, 0, 0, 0, 0)
5.上一步明显可以随便维护,比如树状数组。但是这里有个问题。题目要求输出每首歌的歌唱家。所以用vector维护那些值分别是哪个歌唱家贡献的。
code:
const int N = 100111;
struct point{
int a, b, c, d;
}p1[N], p2[N];
int n, m, cnt;
int li[N<<2];//用于离散化
LL C[N<<2];//树状数组
int ans[N];
vector<PII >vec[N<<2];
bool cmp(point x, point y){
return x.a < y.a;
}
inline int low_bit(int x){
return (x & (-x));
}
void update(int x, int v){
for(; x <= cnt; x += low_bit(x)) C[x] += v;
}
LL ask(int x){
LL ans = 0;
for(; x > 0; x -= low_bit(x)) ans += C[x];
return ans;
}
void add(int id){//加入歌唱家的信息
update(p2[id].b, p2[id].c);
vec[p2[id].b].PB(MP(p2[id].d, p2[id].c));
}
int mid_find(int fr){
LL pre = ask(fr - 1);
if(ask(cnt) - pre <= 0) return -1;
int L = fr, R = cnt;
while(L < R){
int mid = (L + R) >> 1;
LL val = ask(mid) - pre;
if(val == 0) L = mid + 1;
else {
val = ask(mid - 1) - pre;
if(val == 0) return mid;
else R = mid - 1;
}
}
return L;
}
int query(int id){
int mi = mid_find(p1[id].b);
if(mi == -1) return -1;
update(mi, -1);
PII tmp = vec[mi][vec[mi].size()-1];
vec[mi].pop_back();
ans[p1[id].c] = tmp.fi;
tmp.se--;
if(tmp.se > 0) vec[mi].PB(tmp);
}
int main(){
while(scanf("%d", &n) != EOF){
cnt = 0;
for(int i = 1; i <= n; ++i){
scanf("%d%d", &p1[i].a, &p1[i].b);
p1[i].c = i;
li[cnt++] = p1[i].a;
li[cnt++] = p1[i].b;
}
scanf("%d", &m);
for(int i = 1; i <= m; ++i){
scanf("%d%d%d", &p2[i].a, &p2[i].b, &p2[i].c);
if(p2[i].c > n) p2[i].c = n;
p2[i].d = i;
li[cnt++] = p2[i].a;
li[cnt++] = p2[i].b;
}
sort(li, li +cnt);
int j = 1;
for(int i = 1; i < cnt; ++i){
if(li[i] > li[i-1]) li[j++] = li[i];
}
cnt = j;
for(int i = 0; i <= cnt; ++i) vec[i].clear();
for(int i = 0; i <= cnt; ++i) C[i] = 0;
for(int i = 1; i <= n; ++i){
p1[i].a = lower_bound(li, li + cnt, p1[i].a) - li + 1;
p1[i].b = lower_bound(li, li + cnt, p1[i].b) - li + 1;
}
for(int i = 1; i <= m; ++i){
p2[i].a = lower_bound(li, li + cnt, p2[i].a) - li + 1;
p2[i].b = lower_bound(li, li + cnt, p2[i].b) - li + 1;
}
sort(p1 + 1, p1 + n + 1, cmp);
sort(p2 + 1, p2 + m + 1, cmp);
bool has = 1;
for(int i = 1, j = 0; i <= n; ++i){
while(j + 1 <= m && p2[j+1].a <= p1[i].a){
++j;
add(j);
}
int id = query(i);
if(id == -1) {
has = 0;
break;
}
}
if(has){
puts("YES");
for(int i = 1; i <= n; ++i) printf("%d ", ans[i]);
} else {
puts("NO");
}
}
return 0;
}