题目介绍

给定整数N和K，生成一个长度为N的整数数组，使得每个子数组的元素和都能被K整除。

解题思路

由于题目中要求每个子数组的元素求和能够被K整除，因此可以从K的倍数的角度去考虑。对于N个数来说，如果这N个数能够被K整除，那么这N个数的和也会被K整除。

因此我们可以先生成N个K的倍数，再对于其余的N%K个数进行操作，使得它们加上N%K-1个K的倍数之后也能够被K整除。这样就能够保证整个数组中每个子数组的元素和都能够被K整除。

下面给出代码实现。

代码实现

def generate_array(N, K):
    """
    生成一个长度为N的数组，每个子数组的总和可被K整除
    """
    array = [K] * (N // K)
    if N % K != 0:
        extra = (N // K + 1) * K - N
        array += [K - extra] * (N % K - 1)
        array += [K] * (N // K + 1)
    return array

测试样例

>>> generate_array(10, 3)
[3, 3, 3, 6, 6, 6, 3, 3, 3, 3]

>>> generate_array(7, 2)
[2, 2, 2, 2, 2, 2, 2]

>>> generate_array(5, 5)
[5, 5, 5, 5, 5]

以上是本题的详细介绍及代码实现。