n 个小朋友排成一排,从左到右依次编号为 1∼n。
第 i 个小朋友的身高为 hi。
虽然队伍已经排好,但是小朋友们对此并不完全满意。
对于一个小朋友来说,如果存在其他小朋友身高比他更矮,却站在他右侧的情况,该小朋友就会感到不满。
每个小朋友的不满程度都可以量化计算,具体来说,对于第 i 个小朋友:
- 如果存在比他更矮且在他右侧的小朋友,那么他的不满值等于其中最靠右的那个小朋友与他之间的小朋友数量。
- 如果不存在比他更矮且在他右侧的小朋友,那么他的不满值为 −1。
请你计算并输出每个小朋友的不满值。
注意,第 1 个小朋友和第 2 个小朋友之间的小朋友数量为 0,第 1 个小朋友和第 4 个小朋友之间的小朋友数量为 2。
输入格式
第一行包含整数 n。
第二行包含 n 个整数 h1,h2,…,hn。
输出格式
共一行,输出 n 个整数,第i 个整数为第 i 个小朋友的不满值。
数据范围
前 5 个测试点满足2≤n≤5。
所有测试点满足 2≤n≤10^5,1≤hi≤10^9。
输入样例1:
6
10 8 5 3 50 45
输出样例1:
2 1 0 -1 0 -1
输入样例2:
7
10 4 6 3 2 8 15
输出样例2:
4 2 1 0 -1 -1 -1
输入样例3:
5
10 3 1 10 11
输出样例3:
1 0 -1 -1 -1
分析:先观察性质可以发现:从右往左看,当左边一个人比右边一个人高时,左边那个人存在就没有意义了,这样就可以形成一个单调递增的序列
然后从后往前依次枚举,如果遇到比当前栈顶元素还小或者等于的数就输出-1,如果大于栈顶元素就需要二分当前维护的单调栈,注意单调栈是单调递减的,所以二分时需要注意;
距离就是找到的r - i - 1;
最后每次遇到比栈顶元素小的要加到栈中。
// 1 2 3 4 5 6 7 8 9 10
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int w[N],stk[N],ans[N],n;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&w[i]);
int top=0;
for(int i=n;i>=1;i--)
{
if(!top || w[i] <= w[stk[top]]) ans[i]=-1;
else
{
int l=1,r=top;
while(l<r)
{
int mid = (l + r) / 2;
if(w[stk[mid]] < w[i]) r=mid;
else l=mid+1;
}
ans[i] = stk[r] - i - 1;
}
if(!top || w[i] < w[stk[top]] ) stk[++top] = i;
}
for(int i=1;i<=n;i++) cout<<ans[i]<<" ";
return 0;
}
第二种写法:
先建立一个后缀min数组,也就是从后往前看是单调递减的。
为什么建立一个后缀min数组就可以解决问题呢?
首先后缀min数组是从最后往前依次更新的,如果前一个小朋友比后一个小朋友高,那么就把前一个小朋友更新为后一个小朋友的高度,因为一个小朋友想要找到最右边比他矮的,那么刚才那个高的小朋友存在就没什么意义,最后依次枚举找到第一个小于等于小朋友;
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1e5 + 10;
int a[N],h[N];
int ans[N];
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
ans[n]=-1;
h[n]=a[n];
for(int i=n-1;i>=1;i--) h[i]=min(a[i],h[i+1]);
// for(int i=1;i<=n;i++) cout<<h[i]<<" ";
// cout<<endl;
for(int i=1;i<=n-1;i++)
{
int l=i,r=n;
while(l<r)
{
int mid=(l+r+1)/2;
if(h[mid] <= a[i]) l=mid;
else r=mid-1;
}
// cout<<i<<" "<<r<<endl;
ans[i]=r-i-1;
}
for(int i=1;i<=n;i++) cout<<ans[i]<<" ";
return 0;
}