参考博客:https://blog.csdn.net/a664607530/article/details/52909700
线段树的做法:
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>
#include<cmath>
#include<string>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
struct node{
ll l,r,minn;
}a[400001];
ll num[100001],ans[100001];
void build(ll k,ll l,ll r){
a[k].l=l;a[k].r=r;
if(l==r){
scanf("%lld",&num[l]);
a[k].minn=num[l];
return ;
}
ll mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
a[k].minn=min(a[k<<1].minn,a[k<<1|1].minn);
}
//找到pos处的ans值后,更新涉及到位置pos的区间,主要是更新那些最小值目前只num[pos]的区间
//为什么要更新?因为找ans值是从左向右找的,右边的num值与左边的num值无可比性,只与更右边的num值有可比性
void update(ll k,ll l,ll r,ll pos){
if(l==r){
a[k].minn=INF;//将该位置的num值更新为INF
return ;
}
ll mid=(a[k].l+a[k].r)>>1;
if(pos<=mid)update(k<<1,l,mid,pos);
else update(k<<1|1,mid+1,r,pos);
a[k].minn=min(a[k<<1].minn,a[k<<1|1].minn);//递归更新
}
//寻找pos位置的ans值,ans[pos]
void query(ll k,ll l,ll r,ll pos){
if(l==r){
//在此处,l<pos是一定的,因为在pos之前的元素都已经改为的INF
ans[pos]=l-pos-1;//num[l]是小于num[pos]的最靠右的元素
return ;
}
ll mid=(a[k].l+a[k].r)>>1;
if(a[k<<1|1].minn<num[pos])query(k<<1|1,mid+1,r,pos);//先找右区间
else query(k<<1,l,mid,pos);
}
int main(){
ll n;
while(scanf("%lld",&n)!=EOF){
build(1,1,n);
for(int i=1;i<=n;i++){
if(a[1].minn==num[i])ans[i]=-1;//先把最小的筛选出来。
else query(1,1,n,i);
update(1,1,n,i);//每次找出pos的ans,更新
}
for(int i=1;i<n;i++){
printf("%lld ",ans[i]);
}
printf("%lld\n",ans[n]);
}
return 0;
}