大象的年龄--线段树or单调队列加二分

参考博客: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;
}
发布了186 篇原创文章 · 获赞 38 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_43746332/article/details/103434507