牛的视野(单调栈问题)

题目:一群高度不完全相同的牛从左到右站成一排,每头牛只能看见它右边的比它矮的牛的发型,若遇到一头高度大于或等于它的牛,则无法继续看到这头牛和后面的其他牛的发型。给出这些牛的高度,要求每头牛可以看到的牛的数量的和。

解析:若求每头牛能看到的牛的数量,可以参考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);
	}
}

猜你喜欢

转载自blog.csdn.net/Broken_Wave/article/details/82389920