2.18 PAT1007 乙级 python

素数对猜想

import math
x =int(input(''))
num = 0
su_list = [3]
l = 0
def sushu(j,primes):
	updata = math.sqrt(j)+1
	for prime in primes:
		if prime>updata:
			break
		if j%prime == 0:
			return 0
	return 1

for i in range(5,x+1,2):
	if sushu(i,su_list):
		su_list.append(i)
		l = l+1
		if su_list[l]-su_list[l-1] == 2:
			num = num+1
	i = i+2
print(num)

这题我也解了很久。。主要原因是最后一个测试超时的问题，所以目的就是要优化代码。

一开始自己在自己钻，就是想着在判断素数这块加快速度。

比如说n,判断n是不是素数的话，循环从2到n，一个一个除过去，看看n是不是能够整除它们。

优化1：循环改到n/2为止

优化2：循环改到根号n的整数部分为止

优化3：优化2的基础上，再可以只循环奇数，因为偶数都不可能是合数，奇数的因数也不可能是偶数

优化4：循环改成对素数判断成功后不断扩张的素数列表（su_list) 因为一个数如果能被某个合数整除，那必能被这个组成这个合数的质因数整除（吼！竟然能唤起小学数学还是初中数学的记忆~），所以在循环的时候可以去除所有的合数，只循环素数，而素数又正好在前面都判断过，加入到了su_list的列表中！