题目:一群高度不完全相同的牛从左到右站成一排,每头牛只能看见它右边的比它矮的牛的发型,若遇到一头高度大于或等于它的牛,则无法继续看到这头牛和后面的其他牛的发型。给出这些牛的高度,要求每头牛可以看到的牛的数量的和。
解析:若求每头牛能看到的牛的数量,可以参考https://blog.csdn.net/Broken_Wave/article/details/82390882,也是使用单调栈
在这里我们把问题等效于每头牛可以被看到的次数之和来求解
构建一个单调栈,栈底元素最大,从左到右依次读取牛的高度,当当前高度小于栈顶元素时,代表目前栈里的牛都可以看到这头牛,记录目前栈里的元素,然后把当前高度入栈,继续下一个操作
import java.util.Scanner;
import java.util.Stack;
public class Main {
public static void main(String[] arg) {
Scanner sc = new Scanner(System.in);
int n=sc.nextInt();
Stack<Integer> st=new Stack<Integer>();
int sum=0;
//首先压入初始值
st.push(sc.nextInt());
int now=0;
//依次读入高度
for(int i=1;i<n;i++) {
now=sc.nextInt();
//当栈不为空并且栈顶元素小于等于当前高度,栈顶出栈,
while(!st.isEmpty()&&now>=st.peek()) {
st.pop();
}
//栈里的牛应该都是能看到当前牛的,栈的大小即能看到当前牛的数量
sum+=st.size();
st.push(now);
}
System.out.println(sum);
}
}