算法38:子数组的最小值之和(力扣907题)----单调栈

题目:

给定一个整数数组 arr,找到 min(b) 的总和,其中 b 的范围为 arr 的每个(连续)子数组。

示例 1:

输入:arr = [3,1,2,4]
输出:17
解释:
子数组为 [3],[1],[2],[4],[3,1],[1,2],[2,4],[3,1,2],[1,2,4],[3,1,2,4]。 
最小值为 3,1,2,4,1,1,2,1,1,1,和为 17。

示例 2:

输入:arr = [11,81,94,43,3]
输出:444


分析:

很显然,需要分析每个子数组的最小值。而单调栈结构就可以帮助我们找到每个元素左、右侧比自己小的最近位置。也就是说,在一定范围内,数组中每个元素都可以作为子数组的最小值。

假设:

1. 假设数组为1,那么子数组最小值就为1.

2. 假设数组为{1,2} 那么子数组就可以为{1} 和 {1,2}

3. 假设数组为{1,2,3} 那么子数组就可以为{1}、{1,2} 和 {1,2,3}

总结:如果以1为子数组最小值,那么1后面出现的比1大的数,有几个数,子数组就为 n + 1。

数组{1,2,3}中,以1为最小值,那么1后面有2个数比1大,子数组就为 2 + 1 = 3个。

假设数组为{1,2,3}, 现在在最小值1前面加一个数2, 变成{2,1,2,3}.  那么子数组就为

{2,1}、{2,1,2} 、{2,1,2,3}

{1}、{1、2}、{1,2,3}

我们发现,子数组数量是原始数组{1,2,3}的2倍,即 2 * 3 = 6 个。

假设,我们在现有的数组中,前面在添加一个元素3. 变成{3,2,1,2,3}.  那么子数组为:

{3,2,1} 、{3,2,1,2} 、{3,2,1,3}

{2,1}、{2,1,2} 、{2,1,2,3}

{1}、{1、2}、{1,2,3}

我们发现,子数组数量是原始数组{1,2,3}的3倍,即 3 * 3 = 9 个.

总结:如果以1为子数组最小值,那么1前面出现的比1大的数,有几个,那么就是 (N +1)* 原始个数。

 以{3,2,1,2,3}为例子。

如果以1为最小值,1后面有2个数比1大,得到 2+1 = 3; 1前面有2个比1大,得到2+1= 3; 3*3 =9; 如果以1为最小值,那么一共有9个子数组。

如果以下标为1的2为最小值,可得 (1+1)* (0+1) = 2; 即如果以下标为1的2值为最小值,子数组有2个。即 {3,2} 和 {2}

如果以下标为3的位置的2值为最小值,前一个位置为1,即0个。前方0个比自己大的;后面1个3比自己大,可得 (0+1)*(1+1) = 2个;即{2}和{2、3}

依次类推,可以得到全部结果.

package code04.单调栈_01;

import java.util.Stack;

/**
 * 力扣力扣907题: 子数组的最小值
 *  https://leetcode.com/problems/sum-of-subarray-minimums/
 */
public class Code04_SumOfMinValueInArray {

    public int sumSubarrayMins(int[] arr)
    {
        int[][] dp = dp(arr);
        //long比int能存更长的数据
        long ans = 0;

        for (int i = 0; i < arr.length; i++) {
            //当前数
            int cur = arr[i];
            //左侧小于等于当前数的个数
            int left = i - dp[i][0];
            //右侧小于等于当前数的个数
            int right = dp[i][1] - i;

            ans +=  left * right * (long)cur;
            ans %= 1000000007;
        }

        return (int) ans;
    }

    //单调栈,统计出每个位置左、右侧比当前数小的位置
    public int[][] dp(int[] arr)
    {
        if(arr == null || arr.length == 0) {
            return null;
        }

        //当前业务需要统计出2列信息, 即左侧比自己小的位置,右侧比自己小的位置
        int[][] dp = new int[arr.length][2];
        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < arr.length; i++)
        {
            while (!stack.isEmpty() && arr[stack.peek()] > arr[i])
            {
                int cur = stack.pop();
                // -1代表不存在左侧比cur下标对应的值更小的值
                int leftIndex = stack.isEmpty() ? -1 : stack.peek();
                dp[cur][0] = leftIndex;
                dp[cur][1] = i;
            }
            //放入下标
            stack.push(i);
        }

        int rightIndex = arr.length;
        //栈中剩余元素,保持单调增
        while (!stack.isEmpty()) {
            int cur = stack.pop();
            // -1代表不存在左侧比cur下标对应的值更小的值
            int leftIndex = stack.isEmpty() ? -1 : stack.peek();
            dp[cur][0] = leftIndex;
            //因为单调增、所有右侧不存在比自己还小的值了
            dp[cur][1] = rightIndex;
        }
        return dp;
    }

    public static void main(String[] args) {

        Code04_SumOfMinValueInArray ss = new Code04_SumOfMinValueInArray();
        int[] aa = {3,1,2,4};
        System.out.println(ss.sumSubarrayMins(aa));
    }
}

虽然测试通过了,但是执行用时99ms,只击败了17%的用户,说明代码不够优秀。

技巧就是,将java原有的Stack替换成自己实现的数组。因为自己实现的数组是固定的,而Stack是需要不断经过扩容的。这样优化,效果很明显。

package code04.单调栈_01;

import java.util.Stack;

/**
 * 力扣力扣907题: 子数组的最小值
 *  https://leetcode.com/problems/sum-of-subarray-minimums/
 */
public class Code04_SumOfMinValueInArray_opt {

    public int sumSubarrayMins(int[] arr)
    {
        int[][] dp = dp(arr);
        //long比int能存更长的数据
        long ans = 0;

        for (int i = 0; i < arr.length; i++) {
            //当前数
            int cur = arr[i];
            //左侧小于等于当前数的个数
            int left = i - dp[i][0];
            //右侧小于等于当前数的个数
            int right = dp[i][1] - i;

            ans +=  left * right * (long)cur;
            ans %= 1000000007;
        }

        return (int) ans;
    }

    //单调栈,统计出每个位置左、右侧比当前数小的位置
    public int[][] dp(int[] arr)
    {
        if(arr == null || arr.length == 0) {
            return null;
        }

        //当前业务需要统计出2列信息, 即左侧比自己小的位置,右侧比自己小的位置
        int[][] dp = new int[arr.length][2];
        int[] stack = new int[arr.length];
        int stackSize = 0;
        for (int i = 0; i < arr.length; i++)
        {
            while (stackSize != 0 && arr[stack[stackSize-1]] > arr[i])
            {
                int cur = stack[--stackSize];
                // -1代表不存在左侧比cur下标对应的值更小的值
                int leftIndex = stackSize == 0 ? -1 : stack[stackSize - 1];
                dp[cur][0] = leftIndex;
                dp[cur][1] = i;
            }
            //放入下标
            stack[stackSize++] = i;
        }

        int rightIndex = arr.length;
        //栈中剩余元素,保持单调增
        while (stackSize != 0) {
            int cur = stack[--stackSize];
            // -1代表不存在左侧比cur下标对应的值更小的值
            int leftIndex = stackSize == 0 ? -1 : stack[stackSize - 1];
            dp[cur][0] = leftIndex;
            //因为单调增、所有右侧不存在比自己还小的值了
            dp[cur][1] = rightIndex;
        }
        return dp;
    }

    public static void main(String[] args) {

        Code04_SumOfMinValueInArray_opt ss = new Code04_SumOfMinValueInArray_opt();
        int[] aa = {3,1,2,4};
        int[][] dp = ss.dp(aa);

        for (int i = 0; i < dp.length; i++) {
            System.out.println("当前下标 :" + i + ", 左侧小值: " + dp[i][0] + ", 右侧小值: " + dp[i][1]);
        }
        System.out.println(ss.sumSubarrayMins(aa));
    }
}

优化完以后,效果很明显: 

最近更新

  1. TCP协议是安全的吗?

    2024-01-28 08:14:05       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-01-28 08:14:05       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-01-28 08:14:05       19 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-01-28 08:14:05       20 阅读

热门阅读

  1. 力扣19-删除链表中倒数第N个节点

    2024-01-28 08:14:05       36 阅读
  2. 【leetcode100-069到073】【栈】五题合集

    2024-01-28 08:14:05       33 阅读
  3. 每日OJ题_算法_二分查找⑧_力扣LCR 173. 点名

    2024-01-28 08:14:05       34 阅读
  4. 【代码分享】

    2024-01-28 08:14:05       28 阅读
  5. 六、MySQL之视图与索引

    2024-01-28 08:14:05       32 阅读
  6. 强化学习 - Trust Region Policy Optimization (TRPO)

    2024-01-28 08:14:05       29 阅读
  7. Kong Upstream

    2024-01-28 08:14:05       30 阅读
  8. 单例模式(五种创建方式)

    2024-01-28 08:14:05       40 阅读
  9. PyTorch 之 nn.Parameter

    2024-01-28 08:14:05       31 阅读