解题思路:
可以说是困难题中为数不多比较简单的了,一眼看题目就是前缀和的变形问题,这里是以k为target,比k小-1,比k大加1,来记录前缀和,包含k的子数组满足要求的条件即该子数组范围的前缀和为0或者1,进一步来说,就是起始位置和终止位置的前缀和相等,再进一步说,就是对于k右边位置的数的前缀和在左边是否有相同的或者小1的(偶数情况),这样任务就变为先把k左边的前缀和用map记录下来,然后在k右边寻找Map中前缀和相等的或者小1的,加起来即可,代码如下:
class Solution {
public:
int countSubarrays(vector<int>& nums, int k) {
int n = nums.size();
int res = 0;
int index = n;
int sum = 0;
unordered_map<int, int> mp;
mp[0] = 1;
for(int i = 0; i < n; i ++) {
if(nums[i] < k) {
sum -= 1;
} else if(nums[i] > k) {
sum += 1;
} else {
index = i;
}
if(i < index) {
mp[sum] ++;
} else{
int count1 = mp[sum];
int count2 = mp[sum - 1];
res = res + count1 + count2;
}
}
return res;
}
};
PS:这道题一开始我就想复杂了,思路还是这个思路,但是我用Map的vector统计了所有的前缀和,然后在比较的时候判断vector中是否在k的右边,属于是画蛇添足了,果然代码在最后一个用例超时了,故记录下来以防止日后忘记。