[力扣 2488] 关于数组题目的求解思路

题目如下所示：

分析题目，如果让 k 作为一个子数组的中位数，这个子数组需要满足的性质为：

1. 比 K 小的元素和比 K 大的元素的数量的差必须为0（K在奇数数组中间），或者比K大的元素要比K小的元素多1（K在偶数数组左边）
2. K必须在子数组中（子区间的有边界，必须从K及其之后开始计算）

所以我们可以将原数组中比K大的数字设为1，比K小的数字设为-1，相等的设为0，这样可以用前缀和的性质来判断区间内的情况。

如果出现了两个下标对应的前缀和相等，那么就可以说明这两个下标之间的元素比k大的和比k小的元素数量相等。

// 用hash来记录每种前缀和出现的次数
Ans += hash[now_sum];
// 当前下标累计的前缀和，如果在之前这个前缀和出现了n次，就说明以当前下标为结束位置，
// 有n个不同的子区间可以满足让k作为中位数

如果后面一个下标的前缀和比前面一个下标的前缀和大1，就可以说明这个区间里面，比K大的数比比K小的数多1个，这两种情都符合K作为子数组中中位数的情况。

Ans += hash[now_sum - 1];
// 以现在的下标处的前缀和为准，比这个前缀和小1，则说明，在子区间中，比K大的数比
// 比K小的数多1，也满足条件

必须还要额外解释一下的是，为什么要用hash表来作为次数记录器，而不是普通的数组，他们的区别在于，hash表实际上不会开很大的数组，比如我要存前缀和为1e9的下表，那么hash就只会增加一个键值对，但是普通数组，就必须开1e9大小的数组，非常浪费空间。

为什么 hash[0] = 1，举个例子，如果是 [1] 找 1为中位数的子区间，但是改变后的数组为 [0]，hash[0] = 1才能得到正确结果。

最终的AC代码：

class Solution {
public:
    int countSubarrays(vector<int>& nums, int k) {
        int kIndex = -1;
        int sum = 0;

        unordered_map<int, int> hash;
        hash[0] = 1;

        for(int i = 0; i < nums.size(); i ++) {
            if(nums[i] == k) {
                kIndex = i;
                break;
            }
        }

        int Ans = 0;

        for(int i = 0; i < nums.size(); i ++) {
             if(nums[i] > k) {
                sum += 1;
            }
            else if(nums[i] < k) {
                sum -= 1;
            }
            
            if(i < kIndex) {
                hash[sum]++;
            }
            else {
                Ans += hash[sum] + hash[sum - 1];
            }
        }
        
        return Ans;
    }
};