作者:小迅
来源:力扣(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)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。