整体二分——[Poi2011]Meteors

来源:互联网 发布:买域名和空间 编辑:程序博客网 时间:2024/06/05 01:52

https://www.luogu.org/problem/show?pid=3527
洛谷时限比较小,目前超时一个点;
正在联系kkk;
我们要求对于每一个国家,它最早完成时间;
那么我们二分完成时间,并且把答案在这个二分区间内的国家也一起进入二分;
能不用线段树就不用;
区间加法可以用前缀和实现的;

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define Ll long longusing namespace std;const int N=300005;struct cs{int to,nxt;}a[N];struct cz{int l,r,v;}c[N];int head[N],ll;int p[N],o[N],oo[N],ans[N];bool ok[N];Ll f[N];int n,m,x,t,nn;void init(int i,int v){for(;i<=m;i+=i&-i)f[i]+=v;}Ll outit(int i){Ll ans=0;for(;i;i-=i&-i)ans+=f[i];return ans;}void add(int k,int kk){    int l=c[k].l,r=c[k].r,v=c[k].v*kk;    if(r>=l)init(l,v),init(r+1,-v);else    init(1,v),init(r+1,-v),init(l,v);}void er(int L,int R,int l,int r){    if(r<l)return;    if(L==R){for(int i=l;i<=r;i++)ans[o[i]]=L;return;}    int mid=L+R>>1;    while(t<mid)add(++t, 1);//一种新的方式,感觉很牛逼     while(t>mid)add(t--,-1);    for(int i=l;i<=r;i++){        int x=o[i]; oo[i]=x; Ll sum=0;        for(int k=head[x];k;k=a[k].nxt){            sum+=outit(a[k].to);            if(sum>=p[x])break;        }        if(sum>=p[x])ok[x]=1;else ok[x]=0;    }    int top=l-1,sum=0;    for(int i=l;i<=r;i++)if(ok[oo[i]])o[++top]=oo[i];    for(int i=l;i<=r;i++)if(!ok[oo[i]])o[++sum+top]=oo[i];    er(L,mid  ,l,top  );    er(mid+1,R,top+1,r);    }int main(){    scanf("%d%d",&n,&m);    for(int i=1;i<=m;i++){        scanf("%d",&x);        a[++ll].to=i;        a[ll].nxt=head[x];        head[x]=ll;    }    for(int i=1;i<=n;i++)scanf("%d",&p[i]);    scanf("%d",&nn);    for(int i=1;i<=nn;i++)scanf("%d%d%d",&c[i].l,&c[i].r,&c[i].v);    c[++nn].l=1;c[nn].r=m;c[nn].v=1e9;    for(int i=1;i<=n;i++)o[i]=i;    er(1,nn,1,n);    for(int i=1;i<=n;i++)if(ans[i]==nn)printf("NIE\n");else printf("%d\n",ans[i]);}
2 0