19徐州网络赛E 线段树加离散化

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

按wi的值建立权值线段树维护值为wi出现的最后位置，对于第i个人的答案，查询线段树[wi+m,max]区间的最大位置pos,令如果pos-i-1小于等于-1则是在i之后不存在大于等于wi+m的人，否则第i个人的答案即为pos-i-1。由于wi范围过大所以需对wi离散化

#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
#define ll long long
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
#define maxn 1000010
int n,m,mx[maxn<<];
ll w[maxn],x[maxn],li[maxn],ans[maxn];
void pushup(int rt)
{
    mx[rt]=max(mx[rt<<],mx[rt<<|]);
}
void update(int L,int c,int l,int r,int rt)
{
    if(l==r)
    {
        mx[rt]=c;
        return ;
    }
    int mid=l+r>>;
    if(L<=mid)update(L,c,ls);
    else update(L,c,rs);
    pushup(rt);
}
int query(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        return mx[rt];
    }
    int mid=l+r>>,ret=;
    if(L<=mid)ret=max(ret,query(L,R,c,ls));
    if(R>mid)ret=max(ret,query(L,R,c,rs));
    return ret;
}
int main()
{
    scanf("%d%d",&n,&m);
    int tot=;
    for(int i=;i<=n;i++)
    {
        scanf("%lld",&w[i]);
        x[++tot]=w[i];
        x[++tot]=w[i]+m;
    }
    sort(x+,x++tot);
    int nx=unique(x+,x++tot)-x-;
    int L;
    for(int i=;i<=n;i++)
    {
        L=lower_bound(x+,x++nx,w[i])-x;
        update(L,i,,nx,);
    }
    for(int i=;i<=n;i++)
    {
        L=lower_bound(x+,x++nx,w[i]+m)-x;
        L=query(L,nx,i,,nx,);
        if(L==-)ans[i]=-;
        else ans[i]=L-i-;
        if(ans[i]<-)ans[i]=-;
    }
    for(int i=;i<n;i++)
    printf("%lld ",ans[i]);
    printf("%lld\n",ans[n]);
    return ;
}