我是靠谱客的博主 执着海燕,最近开发中收集的这篇文章主要介绍LeetCode 992. Subarrays with K Different Integers(K 个不同整数的子数组) 滑动窗口/hard1.Description2.Example3.Solution,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

文章目录

  • 1.Description
  • 2.Example
  • 3.Solution


1.Description

给定一个正整数数组 A,如果 A 的某个子数组中不同整数的个数恰好为 K,则称 A 的这个连续、不一定不同的子数组为好子数组。

(例如,[1,2,3,1,2] 中有 3 个不同的整数:1,2,以及 3。)

返回 A 中好子数组的数目。


2.Example

输入: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].

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

3.Solution

这一题主要难点在于想到:最多包含k个不同整数的子区间数-最多包含k-1个不同整数的子区间数=恰好包含k个不同整数的子区间数
在这里插入图片描述
因此这一题就是求恰好包含k个不同整数的子区间数。

对于代码中的res:为什么可以用新子数组的长度即 【right - left】来表示增加的子数组个数呢?
可以借鉴动态规划的思想,举个例子就好理解了:
当满足条件的子数组从 [A,B,C] 增加到 [A,B,C,D] 时,新子数组的长度为 4,同时增加的子数组为 [D], [C,D], [B,C,D], [A,B,C,D] 也为4。

public class Solution {

    public int subarraysWithKDistinct(int[] A, int K) {
        return atMostKDistinct(A, K) - atMostKDistinct(A, K - 1);
    }

    /**
     * @param A
     * @param K
     * @return 最多包含 K 个不同整数的子区间的个数
     */
    private int atMostKDistinct(int[] A, int K) {
        int len = A.length;
        int[] freq = new int[len + 1];//题中给的A的每一个元素的大小是0到A.length+1
        
        int left = 0;
        int right = 0;
        // [left, right) 里不同整数的个数
        int count = 0;
        int res = 0;
        // [left, right) 包含不同整数的个数小于等于 K
        while (right < len) {
            if (freq[A[right]] == 0) {
                count++;
            }
            freq[A[right]]++;
            right++;

            while (count > K) {
                freq[A[left]]--;
                if (freq[A[left]] == 0) {
                    count--;
                }
                left++;
            }
            // [left, right) 区间的长度就是对结果的贡献,就是从1到K个恰好的和就是最多K的值。
            res += right - left;//因为right在上面已经加一了,因此直接right-left就是左右边的距离了
            //这一步的原因见上边的题解。
        }
        return res;
    }
}

最后

以上就是执着海燕为你收集整理的LeetCode 992. Subarrays with K Different Integers(K 个不同整数的子数组) 滑动窗口/hard1.Description2.Example3.Solution的全部内容,希望文章能够帮你解决LeetCode 992. Subarrays with K Different Integers(K 个不同整数的子数组) 滑动窗口/hard1.Description2.Example3.Solution所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(57)

评论列表共有 0 条评论

立即
投稿
返回
顶部