LeetCode:子数组的最小值之和

一开始的思路是打算先求出所有子数组,然后排序子数组,最后相加,然而运算太复杂,会超出时间限制

sumSubarrayMins (arr) {
      let i, j, subarr, allarr = [], res = 0
      
      for (i = 1; i < arr.length + 1; i++) {
        for (j = 0; j < arr.length - i + 1; j++) {
          subarr = arr.slice(j, j+i)
          allarr.push(subarr)
        }
      }
      
      for (let k = 0; k < allarr.length; k++) {
        allarr[k].sort((a, b) => { return a - b })
        res += allarr[k][0]
      }

      console.log(allarr)
      console.log(res)
    }

后来改变思路,可以试着求每个arr[i]在子数组中为最小值的长度(也就是个数)

例如:

[3,1,2,4] 的最小值为 3,1,2,4,1,1,2,1,1,1,

3有1个、1有6个、2有2个、4有1个

3*1 + 1*6 + 2*2 + 4*1 =17

sumSubarrayMins2 (a) {
      let i,l,r,res = 0,mod=1000000007
      for (i = 0; i < a.length; i++) {
        l = i - 1
        for (; l >= 0 && a[i] < a[l]; l--);
        
        r = i + 1
        for (; r < a.length && a[i] <= a[r]; r++);

        res += (i - l) * (r - i) * a[i]
        
      }
      console.log(res % mod)
    }
发布了18 篇原创文章 · 获赞 3 · 访问量 823

猜你喜欢

转载自blog.csdn.net/qq_16049879/article/details/87347631