整体二分——BZOJ2527/Luogu3527 [POI2011]MET-Meteors

来源:互联网 发布:大战神张飞进阶数据 编辑:程序博客网 时间:2024/05/16 05:59

题面:洛谷3527 BZOJ2527
这题首先可以想到二分答案但是呢有很多点需要二分
同时每个点的二分状态都差不多
那么我们就一起二分(整体二分)
每次二分出答案以后对于每个国家计算有几颗流星
这个树状数组和线段树都可以维护(我用树状数组调了一个上午QAQ)
然后超过希望总数的点放到左边去,没有的放到右边去
然后分别继续二分
对然后统计答案即可

#include<bits/stdc++.h>using namespace std;typedef long long lll;int n,m,k,nedge=0,p[300001],nex[300001],head[300001];int v[300001],L[300001],R[300001],t[300001],ans[300001];lll p1[300001],p2[300001],rp=0,f[300001];bool jzq[300001];inline int lowbit(int x){return(x&-x);}inline void add(int x,lll s){for(;x<=m;x+=lowbit(x))f[x]+=s;}inline lll sum(int x){lll ans=0;for(;x>0;x-=lowbit(x))ans+=f[x];return ans;}inline void addedge(int a,int b){p[++nedge]=b;nex[nedge]=head[a];head[a]=nedge;}void jia(int x,int v){    int l=L[x],r=R[x],vv=v*t[x];    if(r>=l)add(r+1,-vv),add(l,vv);    else add(1,vv),add(r+1,-vv),add(l,vv);}void erfen(int l,int r,int ll,int rr){    if(rr<ll)return;    if(l==r){        for(int i=ll;i<=rr;i++)ans[p1[i]]=l;        return;    }    int mid=l+r>>1;    while(rp<mid)jia(++rp,1);while(rp>mid)jia(rp--,-1);    lll s;    for(int i=ll;i<=rr;i++){        p2[i]=p1[i];s=0;        for(int k=head[p1[i]];k;k=nex[k]){            s+=sum(p[k]);            if(s>=v[p1[i]])break;        }        jzq[p1[i]]=(s>=v[p1[i]])?1:0;    }    s=ll-1;int ss=0;    for(int i=ll;i<=rr;i++)if(jzq[p2[i]])p1[++s]=p2[i];    for(int i=ll;i<=rr;i++)if(!jzq[p2[i]])p1[++ss+s]=p2[i];    erfen(l,mid,ll,s);    erfen(mid+1,r,s+1,rr);}int main(){    scanf("%d%d",&n,&m);    for(int i=1;i<=m;i++){        int x;scanf("%d",&x);        addedge(x,i);    }    for(int i=1;i<=n;i++)scanf("%d",&v[i]);    scanf("%d",&k);    for(int i=1;i<=k;i++)scanf("%d%d%d",&L[i],&R[i],&t[i]);    L[++k]=1;R[k]=m;t[k]=1e9;    for(int i=1;i<=n;i++)p1[i]=i;    erfen(1,k,1,n);    for(int i=1;i<=n;i++)if(ans[i]!=k)printf("%d\n",ans[i]);    else puts("NIE");    return 0;}
1 0