数组中每个第K个素数的总和

本篇文章将介绍如何解决以下编程问题：给定一个整数数组和正整数 k，计算数组中每个第 k 个素数的总和。

解题思路

首先，我们需要一个函数来判断一个数是否是素数。这个函数可以通过遍历从 2 到该数的平方根，看是否能整除来判断。具体实现可以参考下面的代码：

def is_prime(n):
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

接下来，我们需要遍历数组，并对每个第 k 个素数进行累加。遍历数组的方法有很多，我们可以使用 for 循环，也可以使用列表推导式等等。这里我们使用普通的 for 循环，具体实现如下：

def sum_of_kth_primes(arr, k):
    prime_count = 0
    result = 0
    for n in arr:
        if is_prime(n):
            prime_count += 1
            if prime_count % k == 0:
                result += n
    return result

其中 prime_count 记录当前已经找到的素数个数，如果当前素数是第 k 个，则将其加入 result 中。最后返回 result 即为答案。

完整代码

def is_prime(n):
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True


def sum_of_kth_primes(arr, k):
    prime_count = 0
    result = 0
    for n in arr:
        if is_prime(n):
            prime_count += 1
            if prime_count % k == 0:
                result += n
    return result

总结

本篇文章介绍了如何解决数组中每个第 k 个素数的总和问题。我们使用一个函数来判断一个数是否为素数，然后遍历数组，并对每个第 k 个素数进行累加，最后返回总和。代码实现简单，算法效率也较高，适用于小型数据集。