📌  相关文章
📜  通过合并元素和添加元素来修改数组,使其仅包含素数。(1)

📅  最后修改于: 2023-12-03 14:58:04.821000             🧑  作者: Mango

修改数组,使其仅包含素数

在编写程序时,有时需要将数组中的元素进行修改,使其只保留素数。以下是一些方法,可以通过合并数组或添加新的素数元素来实现这一操作。

1. 通过合并数组

可以创建一个空的数组,然后遍历原数组中的每个元素,检查其是否为素数,并将其添加到新数组中。

# 定义一个函数来检查素数
def check_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

# 创建一个空的数组
new_array = []

# 遍历原数组中的所有元素
for element in original_array:
    # 如果元素是素数,则将其添加到新数组中
    if check_prime(element):
        new_array.append(element)

# new_array 现在是只包含素数的数组
2. 通过添加素数元素

如果原数组中只包含少量元素,可以直接添加新的素数元素,而无需创建一个新的数组。

# 定义一个函数来检查素数
def check_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

# 对原数组中的每个元素进行检查
for i in range(len(original_array)):
    # 如果元素不是素数,则替换为下一个素数
    if not check_prime(original_array[i]):
        for j in range(original_array[i] + 1, original_array[i] * 2):
            if check_prime(j):
                original_array[i] = j
                break

# original_array 现在是只包含素数的数组

以上是两种不同的方法,可以根据实际情况选择其中一种适用的方法来处理数组。