LeetCode 记录两个关于前缀的数组题

560. 和为 K 的子数组

把连续元素的和转化成了两个前缀和相减：遍历过程中，如果当前下标的前缀和减去k的值等于之前的某个下标前缀和，说明这两个下标之间的部分的和等于k。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> umap;
        umap[0] = 1;
        int sum = 0;
        int count = 0;
        for (int num:nums) {
            sum += num;
            
            if (umap.find(sum - k) != umap.end()) {
                count += umap[sum - k];
            }
            if (umap.find(sum) == umap.end()) {
                umap[sum] = 1;
            }
            else {
                umap[sum] ++;
            }
        }
        return count;
    }
};

 2909. 元素和最小的山形三元组 II

这题如果暴力的话就是，遍历过程中，对每个元素求前缀中最小的元素，求后缀中最小的元素，然后加起来。最后选一个最小的和即可。

优化：对于每个元素之前和之后的最小元素，我们可以以O(N)的时间先求出来，保存为前缀和后缀数组，数组中记录了该下标的前缀中的最小元素和后缀中的最小元素。

class Solution {
public:
    int minimumSum(vector<int>& nums) {
        vector<int> left_min(nums.size());
        vector<int> right_min(nums.size());
        int current_min = INT_MAX;
        for (int i=0; i < nums.size(); ++i) {
            left_min[i] = min(nums[i], current_min);
            current_min = left_min[i];
        }
        
        current_min = INT_MAX;
        for (int i=nums.size()-1; i >=0; --i) {
            right_min[i] = min(nums[i], current_min);
            current_min = right_min[i];
        }
        int result = INT_MAX;
        for (int i=1; i < nums.size() - 1; ++i) {
            int left = left_min[i-1];
            int right = right_min[i+1];
            if (nums[i] <= left || nums[i] <= right) continue;
            result = min(nums[i] + left + right, result);
        }
        return result == INT_MAX?-1:result;
        
    }
};