检查数组是否由每个不同元素的连续重复的子数组组成

介绍

本文将介绍如何检查一个数组是否由每个不同元素的连续重复的子数组组成。具体来说，对于一个长度为n的数组，如果该数组可以被划分成若干个长度为m的子数组，且每个子数组中的元素均相同，那么我们称该数组由每个不同元素的连续重复的子数组组成。

实现

我们可以使用一个指针last_pos来记录每个子数组的结尾，所以下一个子数组的开头应该是last_pos + 1。对于每个子数组，我们可以通过计算子数组中元素的个数来判断其是否由每个不同元素的连续重复的子数组组成。具体来说，我们可以使用一个变量count来计数，如果某个子数组不满足要求，则返回False。函数的实现如下所示：

def check_array(nums):
    n = len(nums)
    last_pos = -1
    for i in range(n):
        if i == n - 1 or nums[i] != nums[i + 1]:
            if last_pos != -1:
                count = i - last_pos + 1
                if count % (last_pos + 1) != 0:
                    return False
            last_pos = i
    return True

测试

我们可以使用以下测试用例来检验函数的正确性：

assert check_array([1]) == True
assert check_array([1, 1]) == True
assert check_array([1, 2]) == True
assert check_array([1, 1, 2, 2, 2, 3, 3, 4, 4, 4]) == True
assert check_array([1, 1, 2, 2, 2, 3, 3, 4, 4, 5]) == False
assert check_array([1, 1, 2, 2, 3, 3, 4, 4, 4]) == False
assert check_array([1, 1, 2, 2, 3, 3, 4, 5, 4]) == False

以上测试用例覆盖了各种不同的情况，包括长度为1的数组、长度为2的数组、正常情况、长度不满足要求的情况等。运行结果表明该函数可以正确地检查数组是否由每个不同元素的连续重复的子数组组成。