LeetCode题目笔记——1566. 重复至少 K 次且长度为 M 的模式

题目描述

给你一个正整数数组 arr，请你找出一个长度为 m 且在数组中至少重复 k 次的模式。

模式 是由一个或多个值组成的子数组（连续的子序列），连续 重复多次但 不重叠 。 模式由其长度和重复次数定义。

如果数组中存在至少重复 k 次且长度为 m 的模式，则返回 true ，否则返回 false 。

示例1：

输入：arr = [1,2,4,4,4,4], m = 1, k = 3
输出：true
解释：模式 (4) 的长度为 1 ，且连续重复 4 次。注意，模式可以重复 k 次或更多次，但不能少于 k 次。

示例2：

输入：arr = [1,2,1,2,1,1,1,3], m = 2, k = 2
输出：true
解释：模式 (1,2) 长度为 2 ，且连续重复 2 次。另一个符合题意的模式是 (2,1) ，同样重复 2 次。

示例3：

输入：arr = [1,2,1,2,1,3], m = 2, k = 3
输出：false
解释：模式 (1,2) 长度为 2 ，但是只连续重复 2 次。不存在长度为 2 且至少重复 3 次的模式。

示例 4：

输入：arr = [1,2,3,1,2], m = 2, k = 2
输出：false
解释：模式 (1,2) 出现 2次但并不连续，所以不能算作连续重复 2 次。

示例 5：

输入：arr = [2,2,2,2], m = 2, k = 3
输出：false
解释：长度为 2 的模式只有 (2,2)，但是只连续重复 2 次。注意，不能计算重叠的重复次数。

提示：

• 2 <= arr.length <= 100
• 1 <= arr[i] <= 100
• 1 <= m <= 100
• 2 <= k <= 100

题目链接

题目难度——简单

方法一：模拟

  首先，把m*k记作一个变量pattern，表示题目描述的这个模式的长度，然后与原数组长度进行比较，如果原数组长度小于pattern，说明它不可能有这个模式，返回false，否则再继续判断。
  考虑一下特殊情况，比如当m=1时，题目就变成了判断数组里是否有连续出现k次的某数。我们可以用一个变量len维护当前遍历到的可以形成模式的子数组长度，初始为m，即默认从出现一次开始，从第m个位置开始遍历，每次判断当前位置i与i-m位置上的元素是否相等，相等就把len+1，不相等就重置为m，直到len==pattern时返回true，或者遍历完返回false。

代码/C++

class Solution {
public:
    bool containsPattern(vector<int>& arr, int m, int k) {
        int n = arr.size(), pattern = m * k;
        // 如果数组长度小于m*k，那不可能存在题目所述的模式
        if(n < pattern){   
            return false;
        }
        int len = m, i;
        // 从m开始，判断相同的数目
        for(i = m; i < n; i++){
            if(arr[i] == arr[i - m]){
                len++;
            }
            else{
                len = m;
            }
            if(len == pattern){
                return true;
            }
        }
        return false;
    }
};

class Solution:
    def containsPattern(self, arr: List[int], m: int, k: int) -> bool:
        n = len(arr)
        pattern = m * k
        if n < pattern:
            return False
        
        length = m
        for i in range(m, n):
            if arr[i] == arr[i - m]:
                length += 1
            else:
                length = m
            
            if length == pattern:
                return True
        
        return False

总结

  只需要一次遍历，所以时间是O(N)，只用到了常量变量，所以空间是O(1)。