LeetCode 2488 统计中位数为K的子数组【前缀和 Map】HERODING的LeetCode之路

解题思路：
可以说是困难题中为数不多比较简单的了，一眼看题目就是前缀和的变形问题，这里是以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的右边，属于是画蛇添足了，果然代码在最后一个用例超时了，故记录下来以防止日后忘记。