bzoj 2527: [Poi2011]Meteors cdq分治+树状数组

题意

有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;
}