bzoj 2527: [Poi2011]Meteors 整体二分

        给每个国家建一个链表,这样分治过程中的复杂度就和序列长度线形相关了,无脑套整体二分就可以。

      （最坑的地方是如果所有位置都是一个国家，那么它的样本个数会爆longlong！！被这个坑了一次，大于p[i]的时候break就行了）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define N 300005
#define int long long
using namespace std;
int n,m,k;
vector<int>a[N];
int c[N];
void add(int x,int y)
{
    for(int i=x;i<=m;i+=(i&(-i)))
    {
        c[i]+=y;
    }
}
int qur(int x)
{
    int ans=0;
    for(int i=x;i;i-=(i&(-i)))
    {
        ans+=c[i];
    }
    return ans;
}
int q[N];
int p[N];
struct node
{
    int l,r,z;
}op[N];
int ans[N];
int tmp[2][N];
void solve(int L,int R,int l,int r)
{
    if(L>R)return ;
    if(l==r)
    {
        for(int i=L;i<=R;i++)
        {
            ans[q[i]]=l;
        }
        return ;
    }
    int mid=(l+r)>>1;
    int cnt1=0,cnt2=0;
    for(int i=l;i<=mid;i++)
    {
        if(op[i].l<=op[i].r)
        {
            add(op[i].l,op[i].z);add(op[i].r+1,-op[i].z);
        }
        else 
        {
            add(1,op[i].z);add(op[i].r+1,-op[i].z);add(op[i].l,op[i].z);
        }
    }
    for(int i=L;i<=R;i++)
    {
        int tp=0;
        for(int j=0;j<a[q[i]].size();j++)
        {
            tp+=qur(a[q[i]][j]);
            if(tp>=p[q[i]])break;
        }
        if(tp>=p[q[i]])tmp[0][++cnt1]=q[i];
        else p[q[i]]-=tp,tmp[1][++cnt2]=q[i];
    }
    for(int i=l;i<=mid;i++)
    {
        if(op[i].l<=op[i].r)
        {
            add(op[i].l,-op[i].z);add(op[i].r+1,op[i].z);
        }
        else 
        {
            add(1,-op[i].z);add(op[i].r+1,op[i].z);add(op[i].l,-op[i].z);
        }
    }
    int l1=L+cnt1-1;
    for(int i=1;i<=cnt1;i++)q[L+i-1]=tmp[0][i];
    for(int i=1;i<=cnt2;i++)q[l1+i]=tmp[1][i];
    solve(L,l1,l,mid);solve(l1+1,R,mid+1,r);
}
signed main()
{
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int t1;scanf("%lld",&t1);
        a[t1].push_back(i);
    }
    for(int i=1;i<=n;i++)scanf("%lld",&p[i]),q[i]=i;
    scanf("%lld",&k);
    for(int i=1;i<=k;i++)
    {
        scanf("%lld%lld%lld",&op[i].l,&op[i].r,&op[i].z);
    }
    solve(1,n,1,k+1);
    for(int i=1;i<=n;i++)
    {
        if(ans[i]!=k+1)printf("%lld
",ans[i]);
        else puts("NIE");
    }
    return 0;
}