首先,需要定义一个函数来判断一个数是否为质数。这个函数可以采用试除法的思想。
代码如下:
def is_prime(num):
if num <= 1:
return False
elif num <= 3:
return True
elif num % 2 == 0 or num % 3 == 0:
return False
else:
i = 5
while i * i <= num:
if num % i == 0 or num % (i + 2) == 0:
return False
i += 6
return True
接下来,需要构建一个函数来找到给定整数上下五个质数。
代码如下:
def find_primes(num):
primes = []
lower = num - 1
upper = num + 1
count = 0
while count < 5:
if is_prime(lower):
primes.append(lower)
count += 1
lower -= 1
if count == 5:
break
if is_prime(upper):
primes.append(upper)
count += 1
upper += 1
return primes
在主函数中,通过调用find_primes函数,并传入一个整数作为参数,即可得到该整数上下5个质数。
例如,输入整数7,程序会输出:[5, 3, 2, 11, 13]
完整代码如下:
def is_prime(num):
if num <= 1:
return False
elif num <= 3:
return True
elif num % 2 == 0 or num % 3 == 0:
return False
else:
i = 5
while i * i <= num:
if num % i == 0 or num % (i + 2) == 0:
return False
i += 6
return True
def find_primes(num):
primes = []
lower = num - 1
upper =