1. 程式人生 > >整體二分——[Poi2011]Meteors

整體二分——[Poi2011]Meteors

https://www.luogu.org/problem/show?pid=3527
洛谷時限比較小,目前超時一個點;
正在聯絡kkk;
我們要求對於每一個國家,它最早完成時間;
那麼我們二分完成時間,並且把答案在這個二分割槽間內的國家也一起進入二分;
能不用線段樹就不用;
區間加法可以用字首和實現的;

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define Ll long long
using 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]); }