题意:
给出一个长度为nnn的数组,令f(i)f(i)f(i)为在区间[i,n][i,n][i,n]内最远的且小于aia_{i}ai的元素的位置与iii位置之间有多少个元素,求f(1),f(2),...,f(n)f(1),f(2),...,f(n)f(1),f(2),...,f(n)
方法:
从右向左维护一个单调递减栈(因为我们是向右找所有小于我们的值):当a[i]<a[s.top()]∣∣s.empty()a[i]<a[s.top()]||s.empty()a[i]<a[s.top()]∣∣s.empty(),入栈,否则不入栈。
当ai<a[s.top()]a_{i}<a[s.top()]ai<a[s.top()],加入栈,并且ans[i]=−1ans[i]=-1ans[i]=−1,假设右边有一个值小于现在的aia_{i}ai,那么我们一定不可以进栈,与维护的栈矛盾,所以右边一定不存在有元素<ai<a_{i}<ai,因此ans[i]=−1ans[i]=-1ans[i]=−1
否则,此时不入栈,栈内的元素都是小于aia_{i}ai的元素,只需要在里面找出来最远的那个即可,因为有单调性,所以可以二分来找他。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,a[3000005],ans[200005],s[200005],cnt=200001;
int main()
{
cin>>n;
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
//找出比a[i]小的且最远的地方是哪里?
for(int i=n;i>=1;i--)
{
if(cnt==200001||a[i]<a[s[cnt]])
{
s[--cnt]=i;
ans[i]=-1;
}
else if(a[i]==a[s[cnt]]) ans[i]=-1;
else
{
int pos=*(lower_bound(s+cnt,s+1+200000,i,[&](int x,int y){
return a[x]<a[y];
})-1);
if(pos==0) pos=s[200000];
ans[i]=pos-i-1;
}
}
for(int i=1;i<=n;i++) printf("%d ",ans[i]);
return 0;
}
该博客介绍了如何使用单调栈解决数组中寻找每个位置上最远小于当前值元素的问题。通过从右向左遍历数组,维护一个单调递减的栈,当遇到比栈顶元素小的数时,更新答案并入栈,否则更新答案为栈中找到的第一个小于当前值的元素的位置。这种方法利用了栈的单调性,有效提高了算法效率。
470

被折叠的 条评论
为什么被折叠?



