题目大意:一群牛站成一排,每头牛能看到他右边比自己矮的牛的头顶,但如果有更高的牛挡住,再往右就看不到了,问每头牛能看到的头顶总和。
相当于给定一组数,求每个数右边第一个不小于自己的数离了多远。有几个想不明白的点终于想通了,后续需要多刷刷单调栈的题。
- 这个题为什么要维护一个单调栈?
相当于排队,当后面来了比我更高的牛的时候,我再等下去也不会看到更多的牛了,所以离开,这个时候我和更高的牛的距离就是我能看到的牛的个数。 - 为什么要使用下标入栈而不是高度?
当使用下标入栈时,假设a在b前面,a比b高,b离开了也不影响后续距离的计数。 - 为什么要保证栈非空?
当处理到最后一头牛时,若这头牛比前面的牛矮,那么前面的牛依然不能知道他能看到多少牛,会留在栈里。最后添加一头牛是保证不会有剩余的牛不知道自己的结果,因此最后一头牛必须是最高的。
附代码
#include <iostream>
#include <stack>
#include <stdio.h>
using namespace std;
int p[80010];
int main(){
int n;
cin>>n;
stack<int> st;
for(int i = 0;i < n;i++) cin>>p[i];
p[n] = 0x3f3f3f3f;
long long res = 0;
for(int i = 0;i <= n;i++){
if(st.empty() || p[i] < p[st.top()]) st.push(i);
else{
while(!st.empty() && p[i] >= p[st.top()]){
int j = st.top();
st.pop();
res += i - j - 1;
}
st.push(i);
}
}
cout<<res<<endl;
return 0;
}