The Preliminary Contest for ICPC Asia Xuzhou 2019 E. XKC's basketball team （线段树）

题目链接：https://nanti.jisuanke.com/t/41387

题目大意：对于给定序列，求出对于每个位置求出比该数大于m的最靠右的位置。

思路：首先对序列进行离散化，然后对于每个数的下标更新到线段树上，然后对于每个位置的数考虑，二分查找比该数大于m的第一个位置到最右端，查找下标的最大值即可。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+;
int n,m;
int w[maxn],x[maxn],Max[maxn<<],ans[maxn];
void pushup(int rt){Max[rt]=max(Max[rt<<],Max[rt<<|]);}
void update(int pos,int val,int l,int r,int rt){
    if(l==r){
        Max[rt]=val;
        return;
    }
    int mid=(l+r)>>;
    if(pos<=mid) update(pos,val,l,mid,rt<<);
    else update(pos,val,mid+,r,rt<<|);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt){
    if(L<=l&&R>=r){
        return Max[rt];
    }
    int ans=,mid=(l+r)>>;
    if(mid>=L) ans=max(ans,query(L,R,l,mid,rt<<));
    if(mid<=R) ans=max(ans,query(L,R,mid+,r,rt<<|));
    return ans;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++) {
        scanf("%d",&w[i]);
        x[i]=w[i];
    }
    sort(x+,x+n+);
    int len=unique(x+,x++n)-x-;
    for(int i=;i<=n;i++){
        int pos=lower_bound(x+,x++len,w[i])-x;
        update(pos,i,,len,);
    }
    for(int i=;i<=n;i++){
        int pos=lower_bound(x+,x++len,w[i]+m)-x;
        ans[i]=query(pos,len,,len,)-i-;;
        if(ans[i]<)ans[i]=-;
    }
    printf("%d",ans[]);
    for(int i=;i<=n;i++)
        printf(" %d",ans[i]);
    printf("\n");
    return ;
}