区间子数组个数

题目描述

给定一个元素都是正整数的数组A ，正整数 L 以及 R (L <= R)。
求连续、非空且其中最大元素满足大于等于L 小于等于R的子数组个数。

样例

输入:
A = [2, 1, 4, 3]
L = 2
R = 3
输出: 3
解释: 满足条件的子数组: [2], [2, 1], [3].
注意:

• L, R 和 A[i] 都是整数，范围在 [0, 10^9]。
• 数组 A 的长度范围在[1, 50000]。

题解

• 定义两个变量left和right标识有效数据的左边界和右边界，初始为0，有效数据指的是≤R的数。
• 定义gap标识有效区域中最右侧的在L~R之间的数，初始为-1
• 如果nums[i]>R,那么left、right分别置为i+1，gap置为-1
• 如果nums[i]<L,那么判断gap是否≥left，如果满足则统计子数组的数量
• 如果nums[i]在L和R之间，更新gap的右边界，并且统计子数组的数量right-left+1，最后有效区域右边界right++
• 时间复杂度为O（N）

public int numSubarrayBoundedMax(int[] A, int L, int R) {
		int count=0;
		int l=0,gap=-1;
		for(int i=0;i<A.length;i++) {
			if(A[i]>R) {
				l=i+1;
				gap=-1;
			}else if(A[i]<L) {
				if(gap>=l)
					count+=gap-l+1;
			}else {
				gap=i;
				count+=gap-l+1;
			}
		}
		return count;
	}