当前位置:网站首页>【Leetcode】2104. Sum of Subarray Ranges

【Leetcode】2104. Sum of Subarray Ranges

2022-08-09 22:01:00 记录算法题解

题目地址:

https://leetcode.com/problems/sum-of-subarray-ranges/

给定一个长 n n n数组 A A A,求 ∑ l ≤ r ( max ⁡ A [ l : r ] − min ⁡ A [ l : r ] ) \sum_{l\le r}(\max A[l:r]-\min A[l:r]) lr(maxA[l:r]minA[l:r])

参考https://blog.csdn.net/qq_46105170/article/details/109734115。代码如下:

class Solution {
    
 public:
  using LL = long long;

  LL subArrayRanges(vector<int>& nums) {
    
    auto less = [](int& x, int& y) {
     return x <= y; };
    auto more = [](int& x, int& y) {
     return x >= y; };
    return calc(nums, more) - calc(nums, less);
  }

  LL calc(vector<int>& v, bool (*comp)(int&, int&)) {
    
    stack<int> stk;
    LL res = 0;
    for (int i = 0; i < v.size(); i++) {
    
      while (stk.size() && comp(v[i], v[stk.top()])) {
    
        int idx = stk.top(); stk.pop();
        int l = idx - (stk.size() ? stk.top() : -1), r = i - idx;
        res += (LL)v[idx] * l * r;
      }
      stk.push(i);
    }

    while (stk.size()) {
    
      int idx = stk.top(); stk.pop();
      int l = idx - (stk.size() ? stk.top() : -1), r = v.size() - idx;
      res += (LL)v[idx] * l * r;
    }

    return res;
  }
};

时空复杂度 O ( n ) O(n) O(n)

原网站

版权声明
本文为[记录算法题解]所创,转载请带上原文链接,感谢
https://blog.csdn.net/qq_46105170/article/details/126246227