[BZOJ2527][Poi2011]Meteors（整体二分+bit）

题目描述

传送门

题解

整体二分最少是哪个流星雨
每一次处理(l,mid)之间的修改然后再判断每一个询问是否满足
线段树的常数太大了…必须用bit才能过…
还有这玩意中间炸long long啊（私）所以只要满足条件就赶紧退出…

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define LL long long
#define N 300005

int n,m,k,x;
int tot,point[N],nxt[N],v[N];
struct hp{int id,num;LL goal;}pt[N];
struct hq{int l,r;LL a;}opt[N];
LL C[N];
int ans[N];

void add(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void change(int loc,LL val)
{
    for (int i=loc;i<=m;i+=i&(-i))
        C[i]+=val;
}
LL query(int loc)
{
    LL ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
        ans+=C[i];
    return ans;
}
int cmp(hp a,hp b)
{
    return a.num<b.num;
}
void dvd(int l,int r,int x,int y)
{
    if (x>y) return;
    int mid=(l+r)>>1;
    for (int i=l;i<=mid;++i)
    {
        if (opt[i].l<=opt[i].r)
            change(opt[i].l,opt[i].a),change(opt[i].r+1,-opt[i].a);
        else
        {
            change(opt[i].l,opt[i].a),change(m+1,-opt[i].a);
            change(1,opt[i].a),change(opt[i].r+1,-opt[i].a);
        }
    }
    if (l==r)
    {
        for (int i=x;i<=y;++i)
        {
            LL t=0;
            for (int j=point[pt[i].id];j;j=nxt[j])
            {
                t+=query(v[j]);
                if (t>=pt[i].goal) {ans[pt[i].id]=l;break;}
            }
            if (t<pt[i].goal) ans[pt[i].id]=-1;
        }
        for (int i=l;i<=mid;++i)
        {
            if (opt[i].l<=opt[i].r)
                change(opt[i].l,-opt[i].a),change(opt[i].r+1,opt[i].a);
            else
            {
                change(opt[i].l,-opt[i].a),change(m+1,opt[i].a);
                change(1,-opt[i].a),change(opt[i].r+1,opt[i].a);
            }
        }
        return;
    }
    int px=0,py=y-x+1;
    for (int i=x;i<=y;++i)
    {
        LL t=0;
        for (int j=point[pt[i].id];j;j=nxt[j])
        {
            t+=query(v[j]);
            if (t>=pt[i].goal) {pt[i].num=++px;break;}
        }
        if (t<pt[i].goal)
        {
            pt[i].goal-=t;
            pt[i].num=++py;
        }
    }
    sort(pt+x,pt+y+1,cmp);

    for (int i=l;i<=mid;++i)
    {
        if (opt[i].l<=opt[i].r)
            change(opt[i].l,-opt[i].a),change(opt[i].r+1,opt[i].a);
        else
        {
            change(opt[i].l,-opt[i].a),change(m+1,opt[i].a);
            change(1,-opt[i].a),change(opt[i].r+1,opt[i].a);
        }
    }
    dvd(l,mid,x,x+px-1);
    dvd(mid+1,r,x+px,y);
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;++i) scanf("%d",&x),add(x,i);
    for (int i=1;i<=n;++i) scanf("%lld",&pt[i].goal),pt[i].id=i;
    scanf("%d",&k);
    for (int i=1;i<=k;++i) scanf("%d%d%lld",&opt[i].l,&opt[i].r,&opt[i].a);
    dvd(1,k,1,n);
    for (int i=1;i<=n;++i)
        if (ans[i]==-1) puts("NIE");
        else printf("%d\n",ans[i]);
}