JAVA程序设计: 区间子数组个数(LeetCode:795)

给定一个元素都是正整数的数组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]。

思路:我们将数组分段,其中的每一段都不包含非常大的元素(也就是大于R的元素),然后我们用每一段能组成的子数组数目减去最大值不在[L,R]之间的子数组数量即可。

class Solution {
    public int numSubarrayBoundedMax(int[] A, int L, int R) {
    	
    	int num1=0,num2=0;
    	long ans=0;
    	
    	for(int i=0;i<A.length;i++) {
    		if(A[i]>=L && A[i]<=R) {
    			ans-=num1*(num1+1)/2;
    			num2++; num1=0;
    		}
    		else if(A[i]<L) {
    			num1++;
    			num2++;
    		}
    		else {
    			ans-=num1*(num1+1)/2;
    			ans+=num2*(num2+1)/2;
    			num1=0; num2=0;
    		}
    	}
    	
    	ans+=num2*(num2+1)/2;
    	ans-=num1*(num1+1)/2;
    	
    	return (int)ans;
    }
} 	

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值