Description

Byteotian Interstellar Union有N個成員國。現在它發現了一顆新的星球，這顆星球的軌道被分為M份（第M份和第1份相鄰），第i份上有第Ai個國家的太空站。

這個星球經常會下隕石雨。BIU已經預測了接下來K場隕石雨的情況。 BIU的第i個成員國希望能夠收集Pi單位的隕石樣本。你的任務是判斷對於每個國家，它需要在第幾次隕石雨之後，才能收集足夠的隕石。

1<=n,m,k<=3*10^5 1<=Pi<=10^9 1<=Ai<10^9

Solution

一開始的想法是每個國家二分答案，這樣顯然會T 考慮整體二分。我們二分一個運算元mid，把答案小於mid和大於mid的國家分成兩類遞迴。區間修改單點查詢就用樹狀陣列就行了

需要注意的是中途累計求和可能會爆LL，如果已經夠了就要break

Code

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define lowbit(x) ((x)&(-(x)))

typedef long long LL;
const int INF=0x3f3f3f3f;
const int N=600005;

struct Q {
 
int l,r; LL v;} q[N];

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

int ans[N],rec[N],a[N];
int stack[N];
int n,m,t;

LL sum[N],c[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return
 
 x*v;
}

void change(int x,LL v) {
	for (;x<=m;x+=lowbit(x)) c[x]+=v;
}

void add(int x,LL v) {
	if (q[x].l<=q[x].r) {
		change(q[x].l,v);
		change(q[x].r+1,-v);
	} else {
		change(1,v);
		change(q[x].r+1,-v);
		change(q[x].l,v);
		change(m+1,-v);
	}
}

LL get(int x) {
	LL ret=0;
	for (;x;x-=lowbit(x)) ret+=c[x];
	return ret;
}

void solve(int l,int r,int L,int R) {
	if (r<l||R<L) return ;
	if (L==R) {
		rep(i,l,r) ans[rec[i]]=L;
		return ;
	}
	int mid=(L+R)>>1;
	for (;t<mid;) {
		t++; add(t,q[t].v);
	}
	for (;t>mid;) {
		add(t,-q[t].v); t--;
	}
	rep(i,l,r) {
		sum[rec[i]]=0;
		for (int j=0;j<vec[rec[i]].size();++j) {
			sum[rec[i]]+=get(vec[rec[i]][j]);
			if (sum[rec[i]]>=a[rec[i]]) break;
		}
	}
	int top=0,wjp;
	rep(i,l,r) {
		if (sum[rec[i]]>=a[rec[i]]) {
			stack[++top]=rec[i];
		}
	} wjp=top;
	rep(i,l,r) {
		if (sum[rec[i]]<a[rec[i]]) {
			stack[++top]=rec[i];
		}
	}
	rep(i,l,r) rec[i]=stack[i-l+1];
	solve(l,l+wjp-1,L,mid);
	solve(l+wjp,r,mid+1,R);
}

int main(void) {
    freopen("data.in","r",stdin);
    freopen("myp.out","w",stdout);
	n=read(),m=read();
	rep(i,1,m) {
		int x=read();
		vec[x].push_back(i);
	}
	rep(i,1,n) a[i]=read(),rec[i]=i;
	int k=read();
	rep(i,1,k) q[i].l=read(),q[i].r=read(),q[i].v=read();
	k++; q[k].l=1,q[k].r=m,q[k].v=INF;
	solve(1,n,1,k);
	rep(i,1,n) {
		if (ans[i]==k) puts("NIE");
		else printf("%d\n", ans[i]);
	}
	return 0;
}