992.K个不同整数的子数组

如题~

给定一个正整数数组 A,如果 A 的某个子数组中不同整数的个数恰好为 K,则称 A 的这个连续、不一定独立的子数组为 好子数组
(例如,[1,2,3,1,2] 中有 3 个不同的整数:1,2,以及 3。
返回 A好子数组 的数目。

示例 1:
输入:A = [1,2,1,2,3], K = 2
输出:7
解释:恰好由 2 个不同整数组成的子数组:[1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].

示例 2:
输入:A = [1,2,1,3,4], K = 3
输出:3
解释:恰好由 3 个不同整数组成的子数组:[1,2,1,3], [2,1,3], [1,3,4].

用了个哈希表,双指针,滑动窗口。

毕竟这月前几天的题几乎一直没离开过滑动窗口。

class Solution {
public:
    
    int windows(vector<int>& A, int K) {// 返回A中最多有K个不同数字的子数组个数
        int n = A.size();
        int left = 0, right = 0;//滑动窗口必备
        int hash[25000] = {0};//来个哈希记录次数
        int x = 0;
        int res = 0;//答案
        while (right < n)
        {
            if (hash[A[right]] == 0)//没出现过的数字
            {
                x++;
            } 
            hash[A[right]]++;// 右指针指向的数字个数++
            if (x <= K)//如果当前不同数字的种类小于K
            {
                res += right - left + 1; 
            }
            else
            {
                while (x > K)// 如果当前不同数字的种类大于K,一直移动左边界
                {
                    if (hash[A[left]] == 1)
                    {
                        x--;
                    }
                    hash[A[left]]--;
                    left++;
                }
                // 退出循环一定是 x == K ,所以这里要再次加一下子数组的个数
                res += right - left + 1;
            }
            // 移动右指针
            right++;
        }
        return res;
    }
    int subarraysWithKDistinct(vector<int>& A, int K) {
        return windows(A, K) - windows(A, K - 1);
    }
};

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值