2019 徐州网络赛 E XKC's basketball team（区间最大值）

链接

题意：

有n个人站成一排，每个人有能力值a[i]，每个人会有一个angry值，是最右边的一个能力值加m大于等于a[i]的人与第i个人之间的人数。

思路：

对能力值和位置二维排序，能力值大优先，位置大优先。排序后，能力值递减，那么就找之前的那些人里面能力值大于等于a[i]+m的那些人里面的最大下标。用树状数组之类的维护一下【a[i]+m，最大值】区间内的最大位置即可，离散化一下更加方便。

代码：

#include <bits/stdc++.h>
using namespace std;
const int N=5e5+5;
struct node{
    int a,id;
    bool operator <(const node &x)const {
        if(a==x.a)return id>x.id;
        return a>x.a;
    }
}p[N];

int ls[N<<1],sz;
int tree[N<<1],num[N<<1];

int lowbit(int x){
    return x&(-x);
}

void updata(int pos,int k)
{
    num[pos]=k;
    for(int i=pos;i<=sz;i+=lowbit(i))
    {
        tree[i]=num[i];
        for(int j=1;j<lowbit(i);j<<=1)
            tree[i]=max(tree[i],tree[i-j]);
    }
}

int query(int l,int r)
{
    int ans=num[r];
    while(true)
    {
        ans=max(ans,num[r]);
        if(l==r)break;
        for(--r;r-l>=lowbit(r);r-=lowbit(r))
            ans=max(ans,tree[r]);
    }
    return ans;
}

int ans[N];

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&p[i].a);
        ls[i]=p[i].a;
        ls[n+i]=p[i].a+m;
        p[i].id=i;
    }
    sort(ls+1,ls+1+2*n);
    sz=unique(ls+1,ls+1+2*n)-ls-1;
    sort(p+1,p+1+n);
    for(int i=1;i<=n;i++){
        int tmp=p[i].a+m;
        p[i].a=lower_bound(ls+1,ls+1+sz,p[i].a)-ls;
        updata(p[i].a,p[i].id);
        int tp=query(lower_bound(ls+1,ls+1+sz,tmp)-ls,sz);
        if(!tp){
            ans[p[i].id]=-1;
        }
        else{
            if(tp<p[i].id)ans[p[i].id]=-1;//判断最大位置比当前这个人的要小
            else ans[p[i].id]=tp-p[i].id-1;
        }
    }
    for(int i=1;i<=n;i++){
        printf("%d%c",ans[i],i==n?'\n':' ');
    }
    return 0;
}