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
解题:
找符合条件(连续重复)的子列,设置其起始位置为l
对于每一个起始,我们挨个判断后面的元素是否符合要求。
我们记后面的元素与起始的相对位置为offset
,判断条件arr[l + offset] != arr[l + offset % m]
(下标借助了同余来计算),最后找到符合条件的子列
很神奇的是,这里它老老实实的对起始位置后每一个元素进行判断,而并没有选择直接将数组划分为长为m
的子列来判断。在相同的时间复杂度下,确实挨个判断思路会更清晰,有时不必追求花里胡哨的解题方式,简单直接反而最好。
class Solution:
def containsPattern(self, arr: List[int], m: int, k: int) -> bool:
n = len(arr)
for l in range(n - m * k + 1):
offset = 0
while offset < m * k:
if arr[l + offset] != arr[l + offset % m]:
break
offset += 1
if offset == m * k:
return True
return False