题意:
有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;
}