LeetCode · Pregunta diaria · 2488. Mediana estadística K Subarreglo · Hash + Prefijo Suma

Autor: Xiao Xun

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

Fuente: Código Leet

Los derechos de autor pertenecen al autor. Para reimpresión comercial, comuníquese con el autor para obtener autorización, para reimpresión no comercial, indique la fuente.

tema

ejemplo

tren de pensamiento

Título -> Dada una matriz, devuelve el número de subarreglos cuya mediana es k

• La mediana es igual al subconjunto de k, entonces el subconjunto que satisface la condición debe contarse alrededor de la posición de k. Dado que el sujeto sabe que el número mayor que k es uno o igual al número menor que k, por conveniencia de contar, la contribución es mayor que 1, menor que la contribución −1 y la contribución en sí es 0.

• De esta manera, se cuenta la suma del prefijo, el bit k se usa como bit final y se considera el conteo del bit de inicio a la izquierda, y la suma es igual al valor actual y el valor actual menos uno, eso es decir, la suma del intervalo después de k es 0,1.

el código

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）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。