[BZOJ5249][九省联考2018]IIIDX(线段树)

首先有一个显然的贪心，把树建出来然后后序遍历从大到小填数即可。

但是这样在有重复数字的情况下是不行的，如：

 4 2

1 1 1 2

这样贪心答案是1 1 1 2，但正确答案是1 1 2 1。

这就需要对每个数进行“预订”操作。考虑将数从小到大填进树里，显然当前可能填进的节点一定与已经填过的节点相邻，所以我们把这些节点子树都“预订”好，然后找到最靠后的，且不造成上面那个错误的节点填入，最终整棵树就填好了。

具体实现很难讲清楚，还是看代码吧。

#include <cmath>
#include <cstdio>
#include <algorithm>
#define N 500010
#define lson l ,mid ,x << 1
#define rson mid + 1 ,r ,x << 1 | 1
#define rep(i,l,r) for (int i=l; i<=r; i++)
using namespace std;
int a
 ,ans
 ,head
 ,to
 ,nxt
 ,cnt ,si
 ,sum[N << 2];

void add(int x ,int y){ to[++cnt] = y ,nxt[cnt] = head[x] ,head[x] = cnt ,si[x] += si[y]; }

void update(int p ,int a ,int l ,int r ,int x){
sum[x] += a;
if(l == r) return;
int mid = (l + r) >> 1;
if(p <= mid) update(p ,a ,lson);
else update(p ,a ,rson);
}

int find(int k ,int l ,int r ,int x){
if(l == r) return l;
int mid = (l + r) >> 1;
if(sum[x << 1 | 1] < k) return find(k - sum[x << 1 | 1] ,lson);
else return find(k ,rson);
}

int main(){
int n ,i ,j ,t ,l ,last = 1;
double k; scanf("%d%lf" ,&n ,&k);
rep(i,1,n) scanf("%d" ,&a[i]) ,si[i] = 1;
sort(a + 1 ,a + n + 1);
for(i = n ; i ; i -- ) add((int)floor(i / k) ,i);
for(i = head[0] ; i ; i = nxt[i]) update(to[i] ,si[to[i]] ,1 ,n ,1);
for(i = 1 ; i <= n ; i = last){
while(last <= n && a[i] == a[last]) last ++ ;
for(j = last - i ; j ; j -- ){
t = find(j ,1 ,n ,1) ,ans[t] = a[i] ,update(t ,-si[t] ,1 ,n ,1);
for(l = head[t] ; l ; l = nxt[l]) update(to[l] ,si[to[l]] ,1 ,n ,1);
}
}
rep(i,1,n) printf("%d " ,ans[i]);
return 0;
}