[BZOJ 2527] [Luogu P3527] [POI2011]MET-Meteors

洛谷传送门

BZOJ传送门

题目描述

Byteotian Interstellar Union有$N$个成员国。现在它发现了一颗新的星球，这颗星球的轨道被分为$M$份（第$M$份和第$1$份相邻），第$i$份上有第$A_i$个国家的太空站。 这个星球经常会下陨石雨。BIU已经预测了接下来$K$场陨石雨的情况。 BIU的第$i$个成员国希望能够收集$P_i$单位的陨石样本。你的任务是判断对于每个国家，它需要在第几次陨石雨之后，才能收集足够的陨石。

输入输出格式

输入格式：

第一行是两个数$N,M$。 第二行有$M$个数，第$i$个数$O_i$表示第$i$段轨道上有第$O_i$个国家的太空站。 第三行有$N$个数，第$i$个数$P_i$表示第$i$个国家希望收集的陨石数量。 第四行有一个数$K$，表示BIU预测了接下来的$K$场陨石雨。 接下来$K$行，每行有三个数$L_i,R_i,A_i$，表示第$K$场陨石雨的发生地点在从$L_i$顺时针到$R_i$的区间中（如果$Li\le Ri$，就是$L_i$,$L_i+1$,…,$R_i$，否则就是$Ri,Ri+1,\dots ,m-1,m,1,\dots ,Li$），向区间中的每个太空站提供$A_i$单位的陨石样本。

输出格式：

$N$行。第$i$行的数$W_i$表示第$i$个国家在第$W_i$波陨石雨之后能够收集到足够的陨石样本。如果到第$K$波结束后仍然收集不到，输出$NIE$。

提示

数据范围： $1\le n,m,k\le 3\ast 10^5$， $1\le P_i\le 10^9$， $1\le A_i<10^9$

感谢@noco 提供的翻译

解题分析

蒟蒻A的第一道整体二分题QAQ…
如果有只有一个国家怎么处理？我们可以二分答案再分别计算， 区间加用树状数组维护差分即可， 复杂度$O(Nlo{g}^2(N))$。
然后我们发现似乎维护所有国家也差不多，就只需要二分陨石雨次数，枚举所有国家计算， 如果未达到要求就向上二分， 达到了就向下二分， 多的过程只有枚举所有国家计算一遍，总复杂度依然是$O(Nlo{g}^2(N))$。
不过需要注意一个国家得到的总陨石数可能爆$longlong$， 注意达到了就$break$掉。

代码如下：

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cctype>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <limits.h>
#define R register
#define W while
#define gc getchar()
#define IN inline
#define MX 300050
#define ll long long
#define lbt(i) (i & -i) 
IN char nc()
{
    static const int buflen=1e6;
    static char buf[buflen],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,buflen,stdin),p1==p2)?EOF:*p1++;
}
template <class T>
IN void in(T &x)
{
	x = 0; R char c = nc();
	W (!isdigit(c)) c = nc();
	W (isdigit(c))
	x = (x << 1) + (x << 3) + c - 48, c = nc();
}
int cont, stat, num, deal, lcnt, rcnt;
int req[MX], bel[MX], que[MX], ans[MX], lbuf[MX], rbuf[MX];
ll tree[MX], cnt[MX];
std::vector <int> dom[MX];
struct Event
{
	int tim, lef, rig, val;
}eve[MX << 1];
namespace BIT
{
	IN void add(R int now, const int &del)
	{W (now <= stat) tree[now]+= del, now += lbt(now);}
	IN void modify(R int lef, R int rig, const int &del)
	{
		if(lef <= rig) add(lef, del), add(rig + 1, -del);
		else add(1, del), add(rig + 1, -del), add(lef, del);
	}
	IN ll query(R int now)
	{
		ll ret = 0;
		W (now) ret += tree[now], now -= lbt(now);
		return ret;
	}
}
namespace CDQ
{
	void solve(const int &tl, const int &tr, const int &cl, const int &cr)
	{
		if((tl > tr) || (cl > cr)) return;
		if(tl == tr)
		{
			for (R int i = cl; i <= cr; ++i) ans[que[i]] = tl;
			return;
		}
		int mid = tl + tr >> 1;
		W (deal < mid) ++deal, BIT::modify(eve[deal].lef, eve[deal].rig, eve[deal].val);
		W (deal > mid) BIT::modify(eve[deal].lef, eve[deal].rig, -eve[deal].val), --deal;//像莫队一样操作消除/增加影响
		for (R int i = cl; i <= cr; ++i)
		{
			cnt[que[i]] = 0;
			for (R int j = dom[que[i]].size() - 1; ~j; --j)
			{cnt[que[i]] += BIT::query(dom[que[i]][j]); if(req[que[i]] <= cnt[que[i]]) break;}//注意break掉
		}
		lcnt = rcnt = 0;
		for (R int i = cl; i <= cr; ++i)
		{
			if(req[que[i]] > cnt[que[i]]) rbuf[++rcnt] = que[i];
			else lbuf[++lcnt] = que[i];//达到的甩左边二分边界，没达到的甩右边
		}
		int cmid = cl + lcnt - 1;
		for (R int i = 1; i <= lcnt; ++i) que[i + cl - 1] = lbuf[i];
		for (R int i = 1; i <= rcnt; ++i) que[i + cmid] = rbuf[i];
		solve(tl, mid, cl, cmid);
		solve(mid + 1, tr, cmid + 1, cr);
	}
}
int main(void)
{
	in(cont), in(stat);
	for (R int i = 1; i <= stat; ++i)
	in(bel[i]), dom[bel[i]].push_back(i);
	for (R int i = 1; i <= cont; ++i) in(req[i]), que[i] = i;
	in(num); for (R int i = 1; i <= num; ++i) in(eve[i].lef), in(eve[i].rig), in(eve[i].val);
	eve[num + 1].lef = 1, eve[num + 1].lef = stat, eve[num + 1].val = INT_MAX;//增加一个点， 所有不能达到指定数目的国家答案都为num+1
	deal = 0;
	CDQ::solve(1, num + 1, 1, cont);
	for (R int i = 1; i <= cont; ++i)
	{
		if(ans[i] > num) printf("NIE\n");
		else printf("%d\n", ans[i]);
	}
}