【算法】滑动窗口题单——4.不定长滑动窗口(求子数组个数)

news/2024/5/20 8:59:14 标签: 算法, 滑动窗口, 双指针, 数组, 哈希表, 单调队列

文章目录

  • 前言
  • 2799. 统计完全子数组的数目
    • 解法1——枚举右端点,移动左端点
    • 解法2——枚举左端点,扩展右端点
  • 713. 乘积小于 K 的子数组
  • 1358. 包含所有三种字符的子字符串数目
  • 2302. 统计得分小于 K 的子数组数目
  • 2537. 统计好子数组的数目
  • 2762. 不间断子数组滑动窗口+)

题单来源:https://leetcode.cn/problems/minimum-size-subarray-in-infinite-array/solutions/2464878/hua-dong-chuang-kou-on-shi-jian-o1-kong-cqawc/

前言

这些问题常用的方法有:

  1. 枚举左端点,扩展右端点。
  2. 枚举右端点,收缩左端点。

2799. 统计完全子数组的数目

https://leetcode.cn/problems/count-complete-subarrays-in-an-array/description/

在这里插入图片描述

提示:

1 <= nums.length <= 1000
1 <= nums[i] <= 2000

解法1——枚举右端点,移动左端点

class Solution {
    public int countCompleteSubarrays(int[] nums) {
        Set<Integer> s = new HashSet();
        for (int num: nums) s.add(num);
        
        int n = nums.length, sum = s.size(), ans = 0;
        Map<Integer, Integer> m = new HashMap();
        for (int r = 0, l = 0; r < n; ++r) {
            m.merge(nums[r], 1, Integer::sum);
            while (m.get(nums[l]) > 1) {
                m.merge(nums[l], -1, Integer::sum);
                l++;
            }
            if (m.size() == sum) ans += l + 1;
        }
        return ans;
    }
}

解法2——枚举左端点,扩展右端点

class Solution {
    public int countCompleteSubarrays(int[] nums) {
        Set<Integer> set = new HashSet<>();
        Map<Integer, Integer> map = new HashMap<>();
        for (int num: nums) set.add(num);
        int x = set.size(), n = nums.length, ans = 0;
        for (int l = 0, r = 0; l < n; ++l) {
            while (r < n && map.size() < x) map.merge(nums[r++], 1, Integer::sum);
            if (map.size() == x) ans += n - r + 1;
            map.merge(nums[l], -1, Integer::sum);
            if (map.get(nums[l]) == 0) map.remove(nums[l]);
        }
        return ans;
    }
}

713. 乘积小于 K 的子数组

https://leetcode.cn/problems/subarray-product-less-than-k/description/

在这里插入图片描述

提示:

1 <= nums.length <= 3 * 10^4
1 <= nums[i] <= 1000
0 <= k <= 10^6

枚举右端点,根据窗口内的乘积大小移动左端点。
[ l , r ] [l,r] [l,r]范围内的乘积符合条件时,一共有r-l+1个子数组符合条件计入答案,分别为 [ l + 1 , r ] , [ l + 2 , r ] , . . . , [ r , r ] [l+1,r],[l+2,r],...,[r,r] [l+1,r],[l+2,r],...,[r,r]

class Solution {
    public int numSubarrayProductLessThanK(int[] nums, int k) {
        int n = nums.length, mul = 1, ans = 0;
        for (int l = 0, r = 0; r < n; ++r) {
            mul *= nums[r];
            while (l <= r && mul >= k) mul /= nums[l++];
            ans += r - l + 1;
        }
        return ans;
    }
}

1358. 包含所有三种字符的子字符串数目

https://leetcode.cn/problems/number-of-substrings-containing-all-three-characters/description/

在这里插入图片描述

提示:
3 <= s.length <= 5 x 10^4
s 只包含字符 a,b 和 c ·

使用 c n t [ ] cnt[] cnt[] 数组维护窗口中各个字母的数量。
枚举左端点,拓展右端点,当 [ l , r ] [l,r] [l,r]符合条件时,所有的 [ l , r ] , [ l , r + 1 ] , . . . [ l , n − 1 ] [l,r],[l,r+1],...[l,n-1] [l,r],[l,r+1],...[l,n1]都符合条件,计入答案。

class Solution {
    public int numberOfSubstrings(String s) {
        int[] cnt = new int[3];
        int ans = 0, n = s.length();
        for (int l = 0, r = 0; l < n; ++l) {
            while (r < n && !check(cnt)) cnt[s.charAt(r++) - 'a']++;
            if (check(cnt)) ans += n - r + 1;
            else break;         // 已经不可能有答案了
            cnt[s.charAt(l) - 'a']--;
        }
        return ans;
    }

    public boolean check(int[] cnt) {
        return cnt[0] != 0 && cnt[1] != 0 && cnt[2] != 0;
    }
}

2302. 统计得分小于 K 的子数组数目

https://leetcode.cn/problems/count-subarrays-with-score-less-than-k/description/

在这里插入图片描述

提示:

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

枚举右端点,根据窗口情况移动左端点。
当窗口 [ l , r ] [l,r] [l,r]满足条件时,所有 [ l + 1 , r ] , [ l + 2 , r ] , . . . , [ r , r ] [l+1,r],[l+2,r],...,[r,r] [l+1,r],[l+2,r],...,[r,r]都满足条件。

class Solution {
    public long countSubarrays(int[] nums, long k) {
        long ans = 0, s = 0;
        for (int l = 0, r = 0; r < nums.length; ++r) {
            s += nums[r];
            while (s * (r - l + 1) >= k) s -= nums[l++];
            ans += r - l + 1;
        }
        return ans;
    }
}

2537. 统计好子数组的数目

https://leetcode.cn/problems/count-the-number-of-good-subarrays/description/

在这里插入图片描述

提示:

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

哈希表记录各个元素出现的数量,cnt记录符合条件的下标对数。

枚举窗口的左端点,为了让窗口符合条件扩展右端点。(符合条件之后,所有比当前右端点更靠右的下标作为右端点一定也符合条件。)

class Solution {
    public long countGood(int[] nums, int k) {
        long ans = 0;
        Map<Integer, Integer> m = new HashMap<>();
        int n = nums.length, cnt = 0;   // cnt记录下标对数
        // 枚举左端点,扩展右端点
        for (int l = 0, r = 0; l < n; ++l) {
            while (r < n && cnt < k) {
                m.merge(nums[r], 1, Integer::sum);
                cnt += m.get(nums[r]) - 1;
                r++;
            }
            if (cnt >= k) ans += n - r + 1;
            else break;
            m.merge(nums[l], -1, Integer::sum);
            cnt -= m.get(nums[l]);
        }
        return ans;
    }
}

2762. 不间断子数组滑动窗口+)

https://leetcode.cn/problems/continuous-subarrays/description/

在这里插入图片描述

提示:

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

使用滑动窗口,维护窗口内的最大值和最小值有多种方法。

解法1——TreeMap

TreeMap 会对 key 自动排序,并且有方便的 api 获取最大key和最小key。

class Solution {
    public long continuousSubarrays(int[] nums) {
        long ans = 0;
        int n = nums.length;
        TreeMap<Integer, Integer> s = new TreeMap<>();
        for (int l = 0, r = 0; r < n; ++r) {
            s.merge(nums[r], 1, Integer::sum);
            while (s.lastKey() - s.firstKey() > 2) {
                s.merge(nums[l], -1, Integer::sum);
                if (s.get(nums[l]) == 0) s.remove(nums[l]);
                l++;
            }
            ans += r - l + 1;
        }
        return ans;
    }
}

解法2——单调队列

两个单调队列分别维护窗口中的最大值和最小值

class Solution {
    public long continuousSubarrays(int[] nums) {
        long ans = 0;
        // dq1从大到小,dq2从小到大
        Deque<Integer> dq1 = new ArrayDeque(), dq2 = new ArrayDeque();
        for (int i = 0, j = 0; i < nums.length; ++i) {
            // 处理两个单调队列
            while (!dq1.isEmpty() && nums[i] > nums[dq1.peekLast()]) dq1.pollLast();
            while (!dq2.isEmpty() && nums[i] < nums[dq2.peekLast()]) dq2.pollLast();
            dq1.offerLast(i);
            dq2.offerLast(i);
            while (nums[dq1.peekFirst()] > nums[dq2.peekFirst()] + 2) {
                if (dq1.peekFirst() < dq2.peekFirst()) {
                    j = dq1.peekFirst() + 1;
                    dq1.pollFirst();
                }
                else {
                    j = dq2.peekFirst() + 1;
                    dq2.pollFirst();
                }
            }
            ans += i - j + 1;
        }
        return ans;
    }
}

http://www.niftyadmin.cn/n/5128983.html

相关文章

【C语言】memmove()函数(拷贝重叠内存块函数详解)

&#x1f984;个人主页:修修修也 &#x1f38f;所属专栏:C语言 ⚙️操作环境:Visual Studio 2022 目录 一.memmove()函数简介 1.函数功能 2.函数参数 1>.void * destination 2>.onst void * source 3>.size_t num 3.函数返回值 4.函数头文件 二.memmove()函数…

计算机视觉-光源的目的和作用

光源的目的 机器视觉系统的核心是图像采集和图像处理&#xff0c;而光源则是影响图像水平的重要因素&#xff0c;通过适当的光源照明&#xff0c;使图像中的目标信息与背景信息得到更好的分离&#xff0c;可大大降低图像识别难度&#xff0c;提高系统的精度和可靠性。 对于机器…

2024级199管理类联考之英语二2200核心词汇(第三天)

abstract 抽象的,非具体的 n-摘要ideal adj -理想的 n-理想idealized 理想化的ideology 意识形态,思想体系concept 观念,概念 conception n-构想,怀孕,观念awareness 意识,认识significant 重要的,有意义的 significance n-意义,重要性major v-主修 adj-主要的,成年的 n-成年人…

Java练习题2020-2

"统计1到N的整数中,除了1和自身之外&#xff0c;至少还能被两个数整除的数的个数 输入说明&#xff1a;整数 N(N<10000)&#xff1b; 输出说明&#xff1a;符合条件的数的个数 输入样例&#xff1a;10 输出样例&#xff1a;3 (说明&#xff1a;样例中符合条件的3个数是…

6.卷积神经网络

#pic_center R 1 R_1 R1​ R 2 R^2 R2 目录 知识框架No.1 卷积层一、从全连接到卷积二、卷积层三、代码四、QA No.2 卷积层里的填充和步幅一、填充和步幅二、D2L代码注意点三、QA No.3 卷积层里的多输入和多输出通道一、多输入和多输出通道二、D2L代码注意点三、QA No.4 池化层…

vscode连接服务器一直retry

解决方法 打开vscode控制面板&#xff0c;输入命令remote-ssh: kill vs code server on host 选择一直连接不上的服务器端口 重新连接

实体店做商城小程序如何

互联网电商深入各个行业&#xff0c;传统线下店商家无论产品销售还是服务业&#xff0c;仅靠以往的经营模式&#xff0c;很难拓展到客户&#xff0c;老客流失严重&#xff0c;同时渠道单一&#xff0c;无法实现外地客户购物及线上客户赋能等。 入驻第三方平台有优势但也有不足…

gurobi 安装/license激活 记录

前言&#xff1a;花了好久&#xff0c;被嫌弃惹ww&#xff0c;记录一下踩过的坑 至于为何没安装gurobi也能跑一段时间&#xff0c;直到显示需要license激活&#xff0c;还是未解之迷&#xff0c;需要教教。 首先这是官方给的gurobi license激活教程 我们一步步来复现吧&#…