洛谷P4364 [九省联考2018]IIIDX（线段树）

传送门

题解看得……很……迷？

因为取完一个数后，它的子树中只能取权值小于等于它的数。我们先把权值从大到小排序，然后记$a_i$为他左边（包括自己）所有取完他还能取的数的个数。那么当取完一个点$x$的数之后，我们需要为它子树中的点预留出权值，这些权值肯定在它的左边。但我们不知道它子树中的数会取哪几个数，所以我们就把$x$及其右边的数的$a_i$全都减去$x$的子树大小$size_x$，那么就代表$x$的左边有这么多位置被占据了。那么某一个点$y$要取值的时候，我们只要在线段树上找到最左边的一个点，满足它右边（包括自己）所有的$a_i$都大于等于当前子树的$size$即可，这个在线段树上二分就可以了

然后要注意，父亲给他们预留出了权值，那么在做到儿子的时候把这些预留的空间取出来，也就是把上面的影响消除。父亲只给儿子预留了一次空间，所以消除影响也只需要一次就够了

 //minamoto
 #include<bits/stdc++.h>
 using namespace std;
 inline int read(){
     #define num ch-'0'
     char ch;bool flag=;int res;
     while(!isdigit(ch=getchar()))
     (ch=='-')&&(flag=true);
     for(res=num;isdigit(ch=getchar());res=res*+num);
     (flag)&&(res=-res);
     #undef num
     return res;
 }
 const int N=5e5+;
 int mn[N<<],tag[N<<];
 int n;double k;
 int a[N],ans[N],sz[N],fa[N],cnt[N];
 inline bool cmp(int a,int b){return a>b;}
 #define ls (p<<1)
 #define rs (p<<1|1)
 inline void upd(int p){mn[p]=min(mn[ls],mn[rs]);}
 inline void ppd(int p,int t){mn[p]+=t,tag[p]+=t;}
 inline void pd(int p){ppd(ls,tag[p]),ppd(rs,tag[p]),tag[p]=;}
 void build(int p,int l,int r){
     if(l==r) return (void)(mn[p]=l);
     int mid=(l+r)>>;
     build(ls,l,mid),build(rs,mid+,r),upd(p);
 }
 int query(int p,int l,int r,int k){
     if(l==r) return mn[p]>=k?l:l+;
     int mid=(l+r)>>;if(tag[p]) pd(p);
     return k<=mn[rs]?query(ls,l,mid,k):query(rs,mid+,r,k);
 }
 void update(int p,int l,int r,int ql,int qr,int k){
     if(ql<=l&&qr>=r) return (void)(ppd(p,k));
     int mid=(l+r)>>;if(tag[p]) pd(p);
     if(ql<=mid) update(ls,l,mid,ql,qr,k);
     if(qr>mid) update(rs,mid+,r,ql,qr,k);
     upd(p);
 }
 int main(){
 //    freopen("testdata.in","r",stdin);
     n=read();scanf("%lf",&k);
     for(int i=;i<=n;++i) a[i]=read();
     sort(a+,a++n,cmp);
     for(int i=n-;i;--i)
     cnt[i]=a[i]==a[i+]?cnt[i+]+:;
     for(int i=;i<=n;++i) fa[i]=(int)(i/k),sz[i]=;
     for(int i=n;i;--i) sz[fa[i]]+=sz[i];
     build(,,n);
     for(int i=;i<=n;++i){
         if(fa[i]&&fa[i]!=fa[i-]) update(,,n,ans[fa[i]],n,sz[fa[i]]-);
         int x=query(,,n,sz[i]);
         x+=cnt[x],++cnt[x],x-=(cnt[x]-);ans[i]=x;
         update(,,n,x,n,-sz[i]);
     }
     for(int i=;i<=n;++i) printf("%d ",a[ans[i]]);
     return ;
 }