检查一个数组是否可以分成 K 个连续的非重叠子数组，长度为 M，由单个不同的元素组成

问题描述

给定一个整数数组 nums，判断该数组是否可以被分成 k 个长度为 m 的，且每个子数组都是非重叠的，并且子数组中的每个元素都是相同的。

算法思路

考虑使用滑动窗口的方法来解决此类问题。对于每个子数组，其长度为 m，可以定义一个左右指针，使得每次移动右指针时，窗口长度为 m。在移动指针的过程中，判断当前子数组内的元素是否全部相同，如果是，则加入结果集中，并将左指针移动到下一个子数组的起点。

当加入 k 个子数组后，判断结果集中元素的个数是否等于 n（数组长度），如果是，则说明原数组可以被分成 k 个连续的非重叠子数组，长度为 m，由单个不同的元素组成。

代码实现

def can_divide(nums, k, m):
    n = len(nums)
    if k * m != n:
        return False
    i, j = 0, m
    count = 0
    result = set()
    while j <= n:
        sub = nums[i:j]
        if len(set(sub)) == 1:
            count += m
            result.update(set(sub))
            i, j = j, j + m
        else:
            i += 1
            j += 1
        if count == n:
            return len(result) == 1
    return False

总结

本题需要注意的是，要保证当前子数组的元素全部相同，可以使用 set 消除重复元素。而通过滑动窗口的方法可以保证连续性和非重叠性。因为每次移动左指针时，只移动了 m 个位置，保证了子数组之间的不重叠。而每次移动右指针时，窗口长度为 m，保证了窗口内的元素相邻。