BZOJ 2527: [Poi2011]Meteors

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=300005;
inline int read()
{
char ch=getchar(); int x=0,f=1;
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;
}
vector<int>v
;
struct node{int l,r,c,id;}a
,b
,br
,c
;
int n,m,ans
,k; LL s
;
void add(int x,LL u){for(;x<=m;x+=x&-x)s[x]+=u;}
LL get(int x){LL u=0;for(;x;x-=x&-x)u+=s[x];return u;}
void solve(int l,int r,int lc,int rc)
{
if(l>r)return;
if(lc==rc)
{
for(int i=l;i<=r;i++) if(a[i].id) ans[a[i].id]=lc;
return;
}
int mid=(lc+rc)>>1,bn=0,brn=0,i;
for(i=lc;i<=mid;i++)
{
if(c[i].l>c[i].r)add(1,c[i].c);
add(c[i].l,c[i].c),add(c[i].r+1,-c[i].c);
}
for(i=l;i<=r;i++)
{
LL u=0; int x=a[i].l;
for(int j=0;j<v[x].size();j++)
{
u+=get(v[x][j]);
if(u>=a[i].c)break;
}
if(u>=a[i].c)b[++bn]=a[i];
else a[i].c-=u,br[++brn]=a[i];
}
for(i=lc;i<=mid;i++)
{
if(c[i].l>c[i].r)add(1,-c[i].c);
add(c[i].l,-c[i].c),add(c[i].r+1,c[i].c);
}
for(i=1;i<=bn;i++)a[l+i-1]=b[i];
for(i=1;i<=brn;i++)a[l+bn+i-1]=br[i];
solve(l,l+bn-1,lc,mid);
solve(l+bn,r,mid+1,rc);
}
int main()
{
int i,x;
n=read(),m=read();
for(i=1;i<=m;i++){
x=read(); v[x].push_back(i);
}
for(i=1;i<=n;i++)
a[i]=(node){i,i,read(),i};
k=read();
for(i=1;i<=k;i++)
c[i]=(node){read(),read(),read(),0};
solve(1,n,1,k+1);
for(i=1;i<=n;i++)
if(ans[i]<=k)printf("%d\n",ans[i]);
else printf("NIE\n");
return 0;
}