在第K个数组中找到质数之和

本文主要介绍如何在第K个数组中找到所有质数的和。

质数的定义

一个大于1的自然数，除了1和它本身以外不再有其他的因数，那么它就是质数。

算法思路

• 对于每个数组，遍历每个元素，判断是否为质数。
• 如果是质数，将其加入到结果中。
• 统计所有结果的和。

代码实现

def is_prime(num):
    """
    判断一个数是否为质数
    """
    if num < 2:
        return False
    for i in range(2, int(num ** 0.5) + 1):
        if num % i == 0:
            return False
    return True


def sum_of_prime(k_arrays):
    """
    在第K个数组中找到所有质数的和
    """
    # 初始化结果
    result = 0
    # 遍历每个数组
    for arr in k_arrays:
        # 遍历数组中的每个元素
        for num in arr:
            # 判断是否为质数
            if is_prime(num):
                # 如果是质数，将其加入到结果中
                result += num
    return result

使用示例

k_arrays = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
result = sum_of_prime(k_arrays)
print(result) # 10 + 5 + 7 = 22

以上代码的输出结果为22，说明在这个三个数组中，所有的质数之和为22。

在实际使用时，只需要将需要处理的数组传入sum_of_prime函数中即可，返回结果为所有数组中所有质数之和。