LeetCode 992. K 个不同整数的子数组(双指针)

文章目录

1. 题目

给定一个正整数数组 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].
 
提示:
1 <= A.length <= 20000
1 <= A[i] <= A.length
1 <= K <= A.length

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarrays-with-k-different-integers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 参考官方思路
  • 每次遍历一个右端点 r,以该右端点为结束的满足题意的子数组有多少个
  • 左端点有两个极限位置 l1, l2,[l1, r]刚好有 k 个不同数字,[l2, r] 刚好有 k-1 个不同数字
  • 累加 满足条件的l2 - l1
class Solution {
public:
    int subarraysWithKDistinct(vector<int>& A, int K) {
        int l1 = 0, l2 = 0, r = 0, n = A.size();
        int ct1 = 0, ct2 = 0, ans = 0;
        vector<int> nums1(n+1, 0), nums2(n+1, 0);
        while(r < n)
        {
            if(++nums1[A[r]] == 1)
                ct1++;
            if(++nums2[A[r]] == 1)
                ct2++;
            while(ct1 > K)
            {
                if(--nums1[A[l1]] == 0)
                    ct1--;
                l1++;
            }
            while(ct2 >= K)
            {
                if(--nums2[A[l2]] == 0)
                    ct2--;
                l2++;
            }
            ans += l2-l1;
            r++;
        }
        return ans;
    }
};

40 ms 32.5 MB C++


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
Michael阿明

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页