查询区间 [ id-k,id+k] 小于 val 的个数 num , 再在该区间查询第 num 大的数。
从小到大做,则每次只要找一次就行。 O(n*log2n)
#include <bits/stdc++.h>
using namespace std;
const int maxn = 200005;
int sum[maxn*20+5];//sum为森林的不同根结点
int id[maxn],ans[maxn];
struct E {
int num;
int l,r;
} t[maxn*20+5]; //t为tree树
int n,sz,m,a[maxn],b[maxn],cnt;
int q;
int build(int l,int r) {
int now = ++cnt;
int mid = (l+r)>>1;
if(l<r) {
t[now].l = build(l,mid);
t[now].r = build(mid+1,r);
}
return now;
}
int update(int l,int r,int last,int p) {
int now = ++cnt;
t[now].num = t[last].num+1;
t[now].l = t[last].l;
t[now].r = t[last].r;
int mid = (l + r)>>1;
if(l<r) {
if(p<=mid)t[now].l = update(l,mid,t[last].l,p);
else t[now].r = update(mid+1,r,t[last].r,p);
}
return now;
}
int query(int u,int v,int l,int r,int k){//查询区间第k大
if(l==r)return l;
// v 左孩子包含点个数减去 u 左孩子包含点的个数
int tmp = t[t[v].l].num - t[t[u].l].num;
int mid = (l+r)>>1;
if(k<=tmp)return query(t[u].l,t[v].l,l,mid,k);
else return query(t[u].r,t[v].r,mid+1,r,k - tmp);
}
int query2(int u,int v,int l,int r,int k) {//查询区间小于等于k
if(k==0)return 0;
if(l==r)return t[v].num - t[u].num;
int tmp = t[t[v].l].num - t[t[u].l].num;
int mid = (l+r)>>1;
if(k<=mid)return query2(t[u].l,t[v].l,l,mid,k);
else return tmp+query2(t[u].r,t[v].r,mid+1,r, k);
}
inline int read(){
int s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
return s*w;
}
int main() {
int t;
t = read();
while(t--) {
n = read(); m = read();
for(int i=1; i<=n; i++)a[i]=read(),id[a[i]]=i,ans[i] = 0;
cnt = 0;
sum[0] = build(1,n);//建立空树
for(int i=1; i<=n; i++) sum[i] = update(1,n,sum[i-1],a[i]);
for(int i=1;i<=n;i++){
int tp = id[i],val = i,num;
int l,r;
l = max(tp-m,1);
r = min(tp+m,n);
while(1){
num = query2(sum[l-1],sum[r],1,n,val-1);
if(num==0)break;
else{
ans[i]++;
val = query(sum[l-1],sum[r],1,n,num);
if(val==0)break;
else {
ans[i]+=ans[val];
break;
}
}
}
}
for(int i=1;i<n;i++){
printf("%d ",ans[i]+1);
}
printf("%d",ans[n]+1);
printf("\n");
}
return 0;
}
/*
*/