美文网首页
【算法题】2537. 统计好子数组的数目

【算法题】2537. 统计好子数组的数目

作者: 程序员小2 | 来源:发表于2023-04-08 09:05 被阅读0次

    题目:

    给你一个整数数组 nums 和一个整数 k ,请你返回 nums 中 好 子数组的数目。

    一个子数组 arr 如果有 至少 k 对下标 (i, j) 满足 i < j 且 arr[i] == arr[j] ,那么称它是一个 好 子数组。

    子数组 是原数组中一段连续 非空 的元素序列。

    示例 1:

    输入:nums = [1,1,1,1,1], k = 10
    输出:1
    解释:唯一的好子数组是这个数组本身。
    示例 2:

    输入:nums = [3,1,4,3,2,2,4], k = 2
    输出:4
    解释:总共有 4 个不同的好子数组:

    • [3,1,4,3,2,2] 有 2 对。
    • [3,1,4,3,2,2,4] 有 3 对。
    • [1,4,3,2,2,4] 有 2 对。
    • [4,3,2,2,4] 有 2 对。

    提示:

    1 <= nums.length <= 10^5
    1 <= nums[i], k <= 10^9

    java 代码:

    class Solution {
        public long countGood(int[] nums, int k) {
            var cnt = new HashMap<Integer, Integer>();
            long ans = 0;
            int left = 0, pairs = 0;
            for (int x : nums) {
                pairs += cnt.getOrDefault(x, 0);
                cnt.merge(x, 1, Integer::sum); // 移入右端点
                while (pairs - cnt.get(nums[left]) + 1 >= k)
                    pairs -= cnt.merge(nums[left++], -1, Integer::sum); // 移出左端点
                if (pairs >= k) ans += left + 1;
            }
            return ans;
        }
    }
    
    

    相关文章

      网友评论

          本文标题:【算法题】2537. 统计好子数组的数目

          本文链接:https://www.haomeiwen.com/subject/upwmddtx.html