LWC 74: 795. Number of Subarrays with Bounded Maximum

用户1147447

发布于 2019-05-26 00:21:07

2600

发布于 2019-05-26 00:21:07

文章被收录于专栏：机器学习入门

LWC 74: 795. Number of Subarrays with Bounded Maximum

Problem:

We are given an array A of positive integers, and two positive integers L and R (L <= R). Return the number of (contiguous, non-empty) subarrays such that the value of the maximum array element in that subarray is at least L and at most R.

Example :

Input: A = [2, 1, 4, 3] L = 2 R = 3 Output: 3 Explanation: There are three subarrays that meet the requirements: [2], [2, 1], [3].

Note:

L, R and A[i] will be an integer in the range [0, 10^9].
The length of A will be in the range of [1, 50000].

思路：先关注一波性质，在L和R之间的最大值val符合 val in [L, R]，可以转为：求区间内的任意值x ， x in [L, R]。所以，把数组的每个位置当作起点，如A = [2, 1, 4, 3]:

位置0： [2], [2, 1], [2, 1, 4], [2, 1, 4, 3]
位置1： [1], [1, 4], [1, 4, 3]
位置2： [4], [4, 3]
位置3： [3]

for each subset in each position:
calculate whether the last value in subset <= R

你会发现，考虑位置0，subset[2, 1, 4]中的last value 4时，因为4 > R，所以删除该subset，且连同位置1，2，3中包含4的子集可以一并删除。

代码如下：

    public int numSubarrayBoundedMax(int[] A, int L, int R) {
        return count(A, R) -count(A, L - 1);
    }

    int count(int[] a, int r) {
        int ret = 0;
        int cnt = 0;
        for (int v : a) {
            if (v <= r) cnt ++;
            else cnt = 0;
            ret += cnt;
        }
        return ret;
    }

Python版本：

class Solution(object):
    def numSubarrayBoundedMax(self, A, L, R):
        """
        :type A: List[int]
        :type L: int
        :type R: int
        :rtype: int
        """
        def count(A, R):
            ret = cnt = 0
            for v in A:
                cnt = cnt + 1 if v <= R else 0
                ret += cnt
            return ret

        return count(A, R) - count(A, L - 1)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2018年03月06日，如有侵权请联系 cloudcommunity@tencent.com 删除

element