luogu P3527 [POI2011]MET-Meteors

luogu P3527 [POI2011]MET-Meteors

大意

就是给你一个长度为m的环，环上每个点属于一个国家，一共有n个国家，每次会有陨石掉落在环上的一段点上，每个国家都想要收集一定的陨石，问每个国家最少在第几次陨石掉落时就收集够了

题解

整体二分裸题
首先二分所有的询问，然后把满足要求的丢一边，不满足的丢另一边，然后继续二分下去即可

#include<bits/stdc++.h>
#define lowbit(x) (x & -x)
#define int long long
#define N 1000005
using namespace std;
struct A{
	int nd, id;
}b[N], p1[N], p2[N];
struct Q{
	int l, r, x;
}q[N];
int tree[N];
vector<int> g[N];
void update(int x, int y){
	for(;x < N; x += lowbit(x)) tree[x] += y;
}
int query(int x){
	int ret = 0;
	for(; x; x -= lowbit(x)) ret += tree[x];
	return ret;
}
int ans[N], a[N], n, m, tot;
void ef(int l, int r, int x, int y){//l, r是二分的询问的区间，x,y表示的是b[x...y]的国家
	if(x > y) return;
	if(l == r){
		for(int i = x; i <= y; i ++) ans[b[i].id] = l;
		return;
	}
	int len1 = 0 , len2 = 0;
	int mid = (l + r) >> 1;
	for(int i = l; i <= mid; i ++) update(q[i].l, q[i].x), update(q[i].r + 1, - q[i].x);//二分陨石从l..mid
	for(int i = x; i <= y; i ++){
		int tmp = 0;
		for(int j = 0; j < g[b[i].id].size(); j ++){
			tmp += query(g[b[i].id][j]) + query(g[b[i].id][j] + m);//把每个国家可获得的陨石数统计下来
			if(tmp >= b[i].nd) break;//如果过了就直接退出，可以减枝还可以防止爆long long
		}
		if(tmp >= b[i].nd) p1[++ len1] = b[i];//满足就丢p1
		else p2[++ len2] = b[i], p2[len2].nd -= tmp;//不满足就丢p2,注意要减去l...mid陨石坠落对它的贡献
	}
	for(int i = l; i <= mid; i ++) update(q[i].l, - q[i].x), update(q[i].r + 1, q[i].x);//清空树状数组
	for(int i = 1; i <= len1; i ++) b[i + x - 1] = p1[i];
	for(int i = 1; i <= len2; i ++) b[i + x + len1 - 1] = p2[i];//合并到b中
	ef(l, mid, x, x + len1 -1), ef(mid + 1, r, x + len1, y);//继续二分
}
signed main(){
	scanf("%lld%lld", &n, &m);
	for(int i = 1; i <= m; i ++) scanf("%lld", &a[i]), g[a[i]].push_back(i);
	for(int i = 1; i <= n; i ++) scanf("%lld", &b[i].nd), b[i].id = i;
	scanf("%lld", &tot);
	for(int i = 1; i <= tot; i ++){
		scanf("%lld%lld%lld", &q[i].l, &q[i].r, &q[i].x);
		if(q[i].l > q[i].r) q[i].r += m;//处理环的情况
	}
	q[++ tot].l = 1, q[tot].r = n * 2, q[tot].x = 1e16; //加一个来判断是否不能满足要求
	ef(1, tot, 1, n);
	for(int i = 1; i <= n; i ++) if(ans[i] != tot) printf("%lld\n", ans[i]); else printf("NIE\n");	
	return 0;
}