LeetCode 2488. 统计中位数为 K 的子数组

最新推荐文章于 2023-03-16 20:41:14 发布

Sasakihaise_

最新推荐文章于 2023-03-16 20:41:14 发布

阅读量145

点赞数

分类专栏： LeetCode 文章标签：前缀和哈希

本文链接：https://blog.csdn.net/Sasakihaise_/article/details/129583465

版权

LeetCode 专栏收录该内容

438 篇文章 1 订阅

订阅专栏

该文章介绍了一种使用前缀和与哈希映射的方法来统计数组中中位数为K的子数组数量。通过将小于K的元素视为-1，大于K的视为1，然后计算前缀和，可以有效地找出满足条件的子数组。代码示例展示了如何在Java中实现这一算法。

摘要由CSDN通过智能技术生成

2488. 统计中位数为 K 的子数组

【前缀和 + 哈希】把小于k的当做-1，大于k的当做1，那么通过前缀和计算出的一段区间内得分为0或者1就说明符合要求。

class Solution {
    public int countSubarrays(int[] nums, int k) {
        int n = nums.length, idx = 0, ans = 0;
        int[] s = new int[n + 1];
        for (int i = 1; i <= n; i++) s[i] = s[i - 1] + (nums[i - 1] > k ? 1: (nums[i - 1] == k? 0: -1));
        Map<Integer, Integer> map = new HashMap();
        map.put(0, 1);
        for (int i = 0; i < n; i++) {
            if (nums[i] == k) {
                idx = i + 1;
                break;
            }
        }
        for (int i = 1; i <= n; i++) {
            if (i < idx) map.put(s[i], map.getOrDefault(s[i], 0) + 1);
            else {
                ans += map.getOrDefault(s[i], 0) + map.getOrDefault(s[i] - 1, 0);
            }
        }
        return ans;
    }
}