计算数组中的双素数对

简介

本文介绍了如何计算一个数组中的双素数对，双素数对指的是两个素数之差为2的数对，如(3,5),(11,13)等。

实现思路

1. 首先，需要判断一个数是否为素数。若一个数n不是素数，那么n一定可以表示成n = a * b的方式，其中a和b都不等于1和n。 可知，如果要判断n是否为素数，只需要让1到sqrt(n)之间的数分别去除n即可。如果n不能被任何数整除，则n是素数，否则n不是素数。

2. 其次，需要遍历数组中的所有数，并判断它和数组中其它元素对应的差是否为2。若相差2且两个数都是素数，则该数对为一个双素数对，计数器加1。

代码示例

def is_prime(n:int) -> bool:
    '''判断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 count_twin_primes(arr:List[int]) -> int:
    '''计算arr中的双素数对数量'''
    cnt = 0
    for i in range(len(arr)):
        for j in range(i+1, len(arr)):
            if arr[i] - arr[j] == 2 and is_prime(arr[i]) and is_prime(arr[j]):
                cnt += 1
    return cnt

总结

本文介绍了如何计算一个数组中的双素数对，其中介绍了判断素数和遍历数组的基本思路，并给出了代码实现。对于小数据，该算法的计算效率很高，但对于大数据则需要考虑优化。