把连续元素的和转化成了两个前缀和相减:遍历过程中,如果当前下标的前缀和减去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;
}
};
这题如果暴力的话就是,遍历过程中,对每个元素求前缀中最小的元素,求后缀中最小的元素,然后加起来。最后选一个最小的和即可。
优化:对于每个元素之前和之后的最小元素,我们可以以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;
}
};