LeetCode·每日一题·2488. 统计中位数为 K 的子数组·哈希+前缀和

作者：小迅

链接：https://leetcode.cn/problems/count-subarrays-with-median-k/solutions/2172155/ha-xi-qian-zhui-he-zhu-shi-chao-ji-xiang-hgpu/

来源：力扣（LeetCode）

著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

题目

示例

思路

题意 -> 给定一个数组，返回子数组中位数为 k 的个数

• 中位数等于k的子数组，那么满足条件的子数组肯定围绕着k所在的位置计数。由于题目知道，大于k的与小于k的数量差一或者相等，为了方便计数，大于贡献1，小于贡献−1，本身贡献0.

• 这样就前缀和的方式计数了，k位开始作为结束位，考虑左边起始位的计数，求出和等于当前值与当前值减一的数量，即包含k后区间和为0,1的数量。

代码

typedef struct {//哈希结构
    int key;
    int val;
    UT_hash_handle hh;
} HashItem; 

HashItem *hashFindItem(HashItem **obj, int key) {//查询key
    HashItem *pEntry = NULL;
    HASH_FIND_INT(*obj, &key, pEntry);
    return pEntry;
}

bool hashAddItem(HashItem **obj, int key, int val) {//添加哈希结构
    if (hashFindItem(obj, key)) {
        return false;
    }
    HashItem *pEntry = (HashItem *)malloc(sizeof(HashItem));
    pEntry->key = key;
    pEntry->val = val;
    HASH_ADD_INT(*obj, key, pEntry);
    return true;
}

bool hashSetItem(HashItem **obj, int key, int val) {//设置哈希结构
    HashItem *pEntry = hashFindItem(obj, key);
    if (!pEntry) {
        hashAddItem(obj, key, val);
    } else {
        pEntry->val = val;
    }
    return true;
}

int hashGetItem(HashItem **obj, int key, int defaultVal) {//获取哈希结构
    HashItem *pEntry = hashFindItem(obj, key);
    if (!pEntry) {
        return defaultVal;
    }
    return pEntry->val;
}

void hashFree(HashItem **obj) {//销毁哈希表
    HashItem *curr = NULL, *tmp = NULL;
    HASH_ITER(hh, *obj, curr, tmp) {
        HASH_DEL(*obj, curr);  
        free(curr);             
    }
}

static inline int sign(int num) {//取大小贡献
    if (num == 0) {
        return 0;
    }
    return num > 0 ? 1 : -1;
}

int countSubarrays(int* nums, int numsSize, int k) {
    int kIndex = -1;
    for (int i = 0; i < numsSize; i++) {//找k位置
        if (nums[i] == k) {
            kIndex = i;
            break;
        }
    }
    int ans = 0;
    HashItem *counts = NULL;
    hashAddItem(&counts, 0, 1);//入哈希表
    int sum = 0;
    for (int i = 0; i < numsSize; i++) {//遍历
        sum += sign(nums[i] - k);
        if (i < kIndex) {//小于k的位置
            hashSetItem(&counts, sum, hashGetItem(&counts, sum, 0) + 1);
        } else {//大于k的位置
            int prev0 = hashGetItem(&counts, sum, 0);
            int prev1 = hashGetItem(&counts, sum - 1, 0);//取贡献值
            ans += prev0 + prev1;
        }
    }
    hashFree(&counts);
    return ans;
}



作者：小迅
链接：https://leetcode.cn/problems/count-subarrays-with-median-k/solutions/2172155/ha-xi-qian-zhui-he-zhu-shi-chao-ji-xiang-hgpu/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。