[POI2011]Meteors

[POI2011]Meteors

luogu
SPOJ
考虑整体二分
二分国家(l~r)和流星雨(L~R)
树状数组添加L~mid场流星雨的贡献
满足条件的国家放左边,不满足的放右边,累加贡献cur
有可能有些国家不可能满足
可以补一场inf的流星雨
复杂度:\(O(nlog^2n)\)

#define lb(i) (i&-i)
#define ll long long
#include<bits/stdc++.h>
using namespace std;
const int _=3e5+5;
int re(){
    int x=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return x*w;
}
int n,m,k;
int id[_],a[_],ans[_],f[_],g[_];
ll tmp[_],cur[_],t[_];
vector<int>s[_];
struct Q{int l,r,v;}q[_];
void add(int i,int v){while(i<=m)t[i]+=v,i+=lb(i);}
ll sum(int i){ll res=0;while(i)res+=t[i],i-=lb(i);return res;}
void Add(int l,int r,int v){add(l,v);add(r+1,-v);}
void solve(int l,int r,int L,int R){//l~r:country L~R:meteors
    if(l>r)return;
    if(L==R){
        for(int i=l;i<=r;i++)ans[id[i]]=L;
        return;
    }
    int mid=(L+R)>>1;
    for(int i=L;i<=mid;i++){
        if(q[i].l<=q[i].r)Add(q[i].l,q[i].r,q[i].v);
        else Add(q[i].l,m,q[i].v),Add(1,q[i].r,q[i].v);
    }
    int x=0,y=0;
    for(int i=l;i<=r;i++){
        int u=id[i];tmp[u]=cur[u];
        for(int j=0,sz=s[u].size();j<sz;j++){
            tmp[u]+=sum(s[u][j]);
            if(tmp[u]>=a[u]){f[++x]=u;break;}
        }
        if(tmp[u]<a[u])g[++y]=u,cur[u]+=tmp[u]-cur[u];
    }
    for(int i=L;i<=mid;i++){
        if(q[i].l<=q[i].r)Add(q[i].l,q[i].r,-q[i].v);
        else Add(q[i].l,m,-q[i].v),Add(1,q[i].r,-q[i].v);
    }
    for(int i=l;i<l+x;i++)id[i]=f[i-l+1];
    for(int i=l+x;i<=r;i++)id[i]=g[i-x-l+1];
    solve(l,l+x-1,L,mid);solve(l+x,r,mid+1,R);
}
int main(){
    n=re(),m=re();
    for(int i=1;i<=m;i++)
        s[re()].push_back(i);
    for(int i=1;i<=n;i++)
        a[i]=re(),id[i]=i;
    k=re();
    for(int i=1;i<=k;i++){
        q[i].l=re(),q[i].r=re(),q[i].v=re();
    }
    q[++k]=(Q){1,m,1e9};
    solve(1,n,1,k);
    for(int i=1;i<=n;i++)
        if(ans[i]==k)puts("NIE");
        else printf("%d\n",ans[i]);
    return 0;
}

转载于:https://www.cnblogs.com/sdzwyq/p/9892269.html