n 个小朋友排成一排,从左到右依次编号为 1∼n。
第 i 个小朋友的身高为 hi。
虽然队伍已经排好,但是小朋友们对此并不完全满意。
对于一个小朋友来说,如果存在其他小朋友身高比他更矮,却站在他右侧的情况,该小朋友就会感到不满。
每个小朋友的不满程度都可以量化计算,具体来说,对于第 i 个小朋友:
请你计算并输出每个小朋友的不满值。
注意,第 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;
- }