Leetcode-2537. 统计好子数组的数目

Problem: 2537. 统计好子数组的数目

思路

滑动窗口

解题过程

思路:

使用滑动窗口来维护子数组,并通过组合计数动态调整满足条件的数对数目。具体来说,我们维护一个窗口[l,r],使得窗口内相同元素的对数至少为 k,并计算这样的窗口数目。

关键观察:

  • 当一个元素的频次从 c 增加到 c+1 时,新增加的数对数目为 c(因为新元素可以与之前的 c 个元素形成 c 对)。
  • 当一个元素的频次从 c 减少到 c-1 时,减少的数对数目为 c-1(因为移除的元素与剩余的 c-1 个元素的数对被移除)。

算法步骤:

  • 使用两个指针 l 和 r 维护滑动窗口,使用哈希表 cnt 记录元素频次,使用变量 t 记录窗口内的数对数目。
  • 右指针 r 不断扩展窗口,更新元素频次和数对数目 t。
  • 当 t >= k 时,尝试移动左指针 l 缩小窗口,同时更新数对数目 t,直到窗口不再满足条件。
  • 此时,以 r 结尾且满足条件的子数组数目为 l(即左端点可以是 0 到 l-1 的任意位置)。

Code

python

python 复制代码
class Solution:
    def countGood(self, nums: List[int], k: int) -> int:
        n = len(nums)
        ans = 0
        l = 0
        cnt = defaultdict(int)  # 记录数组中的元素频次
        t = 0  # 记录此时窗口的满足i<j且nums[i]=nums[j]的对数

        for r, x in enumerate(nums):
            cnt[x] += 1
            if cnt[x] >= 2:
                t += cnt[x] - 1
            while t >= k and l < r:
                cnt[nums[l]] -= 1
                if cnt[nums[l]] >= 1:
                    t -= cnt[nums[l]]
                l += 1
            ans += l

        return ans

c++

cpp 复制代码
class Solution {
public:
    long long countGood(vector<int>& nums, int k) {
        int n = nums.size();
        long long ans = 0;
        int l = 0;
        unordered_map<int, int> cnt;
        int t = 0;

        for (int r = 0; r < n; r++) {
            cnt[nums[r]]++;
            if (cnt[nums[r]] >= 2)
                t += cnt[nums[r]] - 1;

            while (t >= k && l < r) {
                cnt[nums[l]]--;
                if (cnt[nums[l]] >= 1)
                    t -= cnt[nums[l]];
                l++;
            }
            ans += l;
        }
        return ans;
    }
};

复杂度

  • 时间复杂度: O(n)
  • 空间复杂度: O(n),用哈希表存储元素频次。
相关推荐
weixin_582470172 分钟前
GS-IR:3D 高斯喷溅用于逆向渲染
人工智能·算法
Lin9成24 分钟前
机器学习集成算法与K-means聚类
算法
JNU freshman1 小时前
算法 之 拓 扑 排 序
数据结构·算法
NAGNIP1 小时前
DeepSeek-R1 架构解析:如何通过强化学习激发大模型的推理能力?
算法
小指纹1 小时前
河南萌新联赛2025第(六)场:郑州大学
java·开发语言·数据结构·c++·算法
爱coding的橙子1 小时前
每日算法刷题Day63:8.19:leetcode 堆6道题,用时1h50min
算法·leetcode·职场和发展
岁忧2 小时前
(nice!!!)(LeetCode 每日一题) 1277. 统计全为 1 的正方形子矩阵 (动态规划)
java·c++·算法·leetcode·矩阵·go·动态规划
Davis_12193 小时前
代码随想录算法训练营27天 | 56. 合并区间、738.单调递增的数字、968.监控二叉树(提高)
数据结构·c++·算法·leetcode·贪心算法
闻缺陷则喜何志丹3 小时前
【倍增 桶排序】后缀数组
c++·算法·倍增·桶排序·后缀数组·lcp·后缀树
Korloa4 小时前
表达式(CSP-J 2021-Expr)题目详解
c语言·开发语言·数据结构·c++·算法·蓝桥杯·个人开发