题目链接 BZOJ

洛谷

每个国家的答案可以二分+求前缀和，于是可以想到整体二分。

在每次Solve()中要更新所有国家得到的值，不同位置的空间站对应不同国家比较麻烦。

注意到每次Solve()其国家数是与区间大小相关的，so根据国家处理，区间更新空间站的值，用vector枚举对应空间站得到每个国家的值。(or边表)

//20048kb	11292ms

//1316ms	24.8MB

#include <cstdio>

#include <cctype>

#include <vector>

#include <algorithm>

#define lb(x) (x&-x)

//#define gc() getchar()

#define MAXIN 300000

#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)

typedef long long LL;

const int N=3e5+5,INF=1e9;

int n,m,K,A[N],Ans[N],q[N],q1[N],q2[N];

std::vector<int> v[N];

char IN[MAXIN],*SS=IN,*TT=IN;

inline int read();

struct Operation

{

	int l,r,v;

	inline void Input(){

		l=read(),r=read(),v=read();

	}

}op[N];

inline int read()

{

	int now=0;register char c=gc();

	for(;!isdigit(c);c=gc());

	for(;isdigit(c);now=now*10+c-'0',c=gc());

	return now;

}

namespace T//区间修改 单点查询

{

	int n;

	LL t[N];

	inline void Modify(int p,int v){

		while(p<=n) t[p]+=v, p+=lb(p);

	}

	inline void Modify_Range(int l,int r,int v)

	{

		Modify(l,v), Modify(r+1,-v);

		if(l>r) Modify(1,v);//Modify(n+1,-v)

	}

	inline void Clear(int p)

	{

		for(; p<=n; p+=lb(p))

			if(t[p]) t[p]=0; else break;

	}

	inline void Clear_Range(int l,int r)

	{

		Clear(l), Clear(r+1);//不能少啊

		if(l>r) Clear(1);

	}

	inline LL Query(int p)

	{

		LL res=0; while(p) res+=t[p], p^=lb(p);

		return res;

	}

}

void Solve(int l,int r,int h,int t)

{

	if(h>t) return;

	if(l==r){

		for(int i=h; i<=t; ++i) Ans[q[i]]=l;

		return;

	}

	int mid=l+r>>1, t1=0, t2=0;

	for(int i=l; i<=mid; ++i) T::Modify_Range(op[i].l,op[i].r,op[i].v);

	for(int i=h,now=q[i]; i<=t; now=q[++i])//now:may be q[n+1]

	{

		LL tmp=0;

		for(int j=0,lim=v[now].size(); j<lim; ++j)

			if((tmp+=T::Query(v[now][j]))>=A[now]) break;//tmp可能爆longlong。。不过也是个剪枝。

		if(tmp>=A[now]) q1[t1++]=now;

		else A[now]-=tmp, q2[t2++]=now;

	}

	for(int i=l; i<=mid; ++i) T::Clear_Range(op[i].l,op[i].r);//T::Modify_Range(op[i].l,op[i].r,-op[i].v);

	//T::Clear(op[i].l), T::Clear(op[i].r);//T::Clear(std::min(op[i].l,op[i].r));//因为有l>r的情况，所以要清空的位置比较麻烦。。这一行的两种都是错的。(而且应是r+1)

	for(int i=0; i<t1; ++i) q[h+i]=q1[i];

	for(int i=0; i<t2; ++i) q[h+t1+i]=q2[i];

	Solve(l,mid,h,h+t1-1), Solve(mid+1,r,h+t1,t);

}

int main()

{

	n=read(),T::n=m=read();

	for(int i=1; i<=m; ++i) v[read()].push_back(i);

	for(int i=1; i<=n; ++i) A[i]=read(), q[i]=i;

	K=read();

	for(int i=1; i<=K; ++i) op[i].Input();

	op[++K]=(Operation){m+1,m+1,0};

	Solve(1,K,1,n);

	for(int i=1; i<=n; ++i) Ans[i]==K?puts("NIE"):printf("%d\n",Ans[i]);

	return 0;

}