找出数组中每个数的右边第一个比它大的数或之间有多少数(单调栈问题)

题目:给出一组数,找出数组中每个数的右边第一个比它大的数

分析:

利用单调栈,从左至右依次压入数据的索引(若直接压数,则还需要一个数组保存栈中元素所对应的数组位置)

当当前元素小于等于栈顶的索引所对应的数组的值,入栈当前索引,否则将栈顶索引出栈,并在栈顶索引所对应的res数组中记录下当前的值。

到最后再检查栈中剩余元素,代表剩余元素右边没有比它大的值,在res对应位置赋值为-1

import java.util.Stack;

public class FindRightMax {
	public static void main(String[] arg) {
		int array[]=new int[] {1,5,3,6,4,8,9,10};
		int res[]=findMax(array);
		for(int num:res) {
			System.out.println(num);
		}
		
	}
	public static int[] findMax(int[] array) {
		int len =array.length;
		Stack<Integer> st = new Stack<Integer>();
		int res[]=new int[len];
		int i=0;
		while(i<len) {
			if(st.isEmpty()||array[i]<=array[st.peek()]) {
				st.push(i);
				i++;
			}else {
				res[st.pop()]=array[i];
				
			}
		}
		while(!st.isEmpty()) {
			res[st.pop()]=-1;
		}
		return res;
	}

}

还有一种问题是求数组中每个数的右边第一个比它大的数之间有多少数,只需要在res中记录当前元素索引与栈顶元素索引的差,最后的残留在栈里的元素可以把对应的res数组设置为-1或者len-st.pop(即右边有多少比当前数小的数,比如牛的视野问题https://blog.csdn.net/Broken_Wave/article/details/82389920

可以使用单调栈来解决这个问题。 具体的做法是,我们先用单调栈找出个数左侧和右侧第一个小于它的的位置,分别用 $left_i$ 和 $right_i$ 表示。这个过程可以参考 [84. 柱状图中最大的矩形](https://leetcode-cn.com/problems/largest-rectangle-in-histogram/)。 然后我们枚举每个数作为区间的最小值,假设当前枚举到的是 $a_i$。那么它能扩展的左边界是 $left_i + 1$,右边界是 $right_i - 1$。于是我们就可以计算以 $a_i$ 为区间最小值的区间的最大值了。 具体的计算公式是:$a_i \times (sum_{left_i+1}^{right_i-1})$,其中 $sum_{left_i+1}^{right_i-1}$ 表示区间 $[left_i+1, right_i-1]$ 中所有的和。 我们枚举所有的 $a_i$,取其中的最大值即可。 以下是参考代码: ``` public static long maxSum(int[] nums) { int n = nums.length; long[] left = new long[n], right = new long[n]; long[] sum = new long[n]; Deque<Integer> stack = new LinkedList<>(); // 计算 left 数组 for (int i = 0; i < n; i++) { while (!stack.isEmpty() && nums[stack.peek()] >= nums[i]) { stack.pop(); } left[i] = stack.isEmpty() ? -1 : stack.peek(); stack.push(i); } // 计算 right 数组 stack.clear(); for (int i = n - 1; i >= 0; i--) { while (!stack.isEmpty() && nums[stack.peek()] >= nums[i]) { stack.pop(); } right[i] = stack.isEmpty() ? n : stack.peek(); stack.push(i); } // 计算 sum 数组 for (int i = 0; i < n; i++) { sum[i] = i == 0 ? nums[i] : sum[i - 1] + nums[i]; } long ans = 0; for (int i = 0; i < n; i++) { long cur = nums[i] * (sum[(int) (right[i] - 1)] - (left[i] == -1 ? 0 : sum[(int) (left[i])])); ans = Math.max(ans, cur); } return ans; } ``` 时间复杂度:$O(n)$。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值