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[N];
struct node{int l,r,c,id;}a[N],b[N],br[N],c[N];
int n,m,ans[N],k; LL s[N];
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;
}