【BZOJ】2527 [Poi2011]Meteors 整体二分+树状数组

xiaoxiao2021-02-28  53

题目传送门

这题的整体二分还是挺好想到的,代码也挺好写的。(比BZOJ3110好写多……)

取当期的陨石波数的一半加入树状数组中,判断当前每个国家拥有的陨石数是否达到需求量,若达到则放到左区间中,也就是减小陨石数,反之则放到右区间,增加陨石数。

orzZH大佬,当讲课人在上面放出了这道题时,ZH大佬瞟了一眼,哼了一声:“这题不是水题吗?还需要讲?”

ZH大佬太强啦QWQ,%%%%%,献上我的膝盖骨——跪膜大佬。

附上AC代码:

#include <cstdio> #include <cctype> #define N 300010 #define lowbit(x) ((x)&(-x)) using namespace std; struct side{int to,nt;}s[N]; struct note{long long w,wz;}a[N],q1[N],q2[N]; struct ask{long long l,r,w;}q[N]; long long n,m,x,h[N],num,p,ans[N],t[N]; inline char nc(){ static char ch[100010],*p1=ch,*p2=ch; return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++; } inline void read(long long &a){ static char ch=nc();int f=1; for (;!isdigit(ch);ch=nc()) if (ch=='-') f=-1; for (a=0;isdigit(ch);a=a*10+ch-'0',ch=nc()); a*=f;return; } inline void add(int x,int y){s[++num]=(side){y,h[x]},h[x]=num;} inline void updata(int x,long long w){for (int i=x; i<=n; i+=lowbit(i)) t[i]+=w;} inline void change(int now,int ww){ for (int i=now; i<=m; i+=lowbit(i)) t[i]+=ww; } inline void updata(int l,int r,long long w){ if (l>r) change(l,w),change(1,w),change(r+1,-w); else change(l,w),change(r+1,-w); } inline long long query(int now){ long long ret=0; for (int i=now; i; i-=lowbit(i)) ret+=t[i]; return ret; } inline void so(int l,int r,int ql,int qr){ if (l>r) return; if (ql==qr){ for (int i=l; i<=r; ++i) ans[a[i].wz]=ql; return; } int mid=(ql+qr)>>1,ll=0,rr=0,len=l; for (int i=ql; i<=mid; ++i) updata(q[i].l,q[i].r,q[i].w); for (int i=l; i<=r; ++i){ long long sum=0; for (int j=h[a[i].wz]; j; j=s[j].nt){ sum+=query(s[j].to); if (sum>=a[i].w) break; } if (sum>=a[i].w) q1[++ll]=a[i]; else q2[++rr]=a[i],q2[rr].w-=sum; } for (int i=1; i<=ll; ++i) a[len++]=q1[i]; for (int i=1; i<=rr; ++i) a[len++]=q2[i]; for (int i=ql; i<=mid; ++i) updata(q[i].l,q[i].r,-q[i].w); return so(l,l+ll-1,ql,mid),so(l+ll,r,mid+1,qr); } int main(void){ read(n),read(m); for (int i=1; i<=m; ++i) read(x),add(x,i); for (int i=1; i<=n; ++i) read(a[i].w),a[i].wz=i; read(p); for (int i=1; i<=p; ++i) read(q[i].l),read(q[i].r),read(q[i].w); so(1,n,1,p+1); for (int i=1; i<=n; ++i) if (ans[i]>p) puts("NIE"); else printf("%d\n",ans[i]); return 0; }

转载请注明原文地址: https://www.6miu.com/read-52892.html

最新回复(0)