题意
有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
分析
一开始的想法是对每个国家二分答案,然后对所有陨石雨建一棵可持久化线段树,即使每次插入的是一个区间,但是我们只要把标记永久化一下就好了,所以每次加入的节点数都是O(logn)的,然后每次查询都是O(logn)。就是不知道空间会不会爆炸。
后来发现可持久化线段树完全可以用一个cdq分治替代掉。
至于每次进行区间加和单点查询的话,为了避免掉线段树的巨大常数,我们只要运用差分+树状数组就好啦。
注意一个国家收集的陨石数量会爆long long.
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=300005;
int n,m,k,tmp[N],num[N],ans[N],ned[N],last[N];
struct edge{int pos,next;}e[N];
struct data{int l,r,k;}a[N];
LL c[N],s[N];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void ins(int x,int y)
{
while (x<=m)
{
c[x]+=y;
x+=x&(-x);
}
}
LL find(int x)
{
LL ans=0;
while (x)
{
ans+=c[x];
x-=x&(-x);
}
return ans;
}
void cdq(int l,int r,int L,int R)
{
if (l>r) return;
if (L==R)
{
for (int i=l;i<=r;i++) ans[num[i]]=L;
return;
}
int mid=(L+R)/2;
for (int i=L;i<=mid;i++)
if (a[i].l<=a[i].r) ins(a[i].l,a[i].k),ins(a[i].r+1,-a[i].k);
else ins(1,a[i].k),ins(a[i].r+1,-a[i].k),ins(a[i].l,a[i].k);
int p=l,q=l;
for (int i=l;i<=r;i++)
{
int x=num[i];
s[x]=0;
for (int j=last[x];j;j=e[j].next) s[x]=min(s[x]+find(e[j].pos),(LL)1e10);
if (s[x]>=ned[x]) q++;
}
for (int i=l;i<=r;i++)
if (s[num[i]]>=ned[num[i]]) tmp[p++]=num[i];
else tmp[q++]=num[i];
for (int i=l;i<=r;i++) num[i]=tmp[i];
cdq(p,r,mid+1,R);
for (int i=L;i<=mid;i++)
if (a[i].l<=a[i].r) ins(a[i].l,-a[i].k),ins(a[i].r+1,a[i].k);
else ins(1,-a[i].k),ins(a[i].r+1,a[i].k),ins(a[i].l,-a[i].k);
cdq(l,p-1,L,mid);
}
int main()
{
n=read();m=read();
for (int i=1;i<=m;i++)
{
int x=read();
e[i].pos=i;e[i].next=last[x];last[x]=i;
}
for (int i=1;i<=n;i++) ned[i]=read(),num[i]=i;
k=read();
for (int i=1;i<=k;i++) a[i].l=read(),a[i].r=read(),a[i].k=read();
cdq(1,n,1,k+1);
for (int i=1;i<=n;i++)
if (ans[i]==k+1) printf("NIE\n");
else printf("%d\n",ans[i]);
return 0;
}