PTA 1007 素数对猜想(Python版)

让我们定义dn​为：dn​=pn+1​−pn​，其中pi​是第i个素数。显然有d1​=1，且对于n>1有dn​是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。

现给定任意正整数N(<105)，请计算不超过N的满足猜想的素数对的个数。

输入格式:

输入在一行给出正整数N。

输出格式:

在一行中输出不超过N的满足猜想的素数对的个数。

输入样例:

20

输出样例:

4

import math


def sushu(a):
    for i in range(3, int(math.sqrt(a)) + 1, 2):
        if a % i == 0:
            return False
    return True


num = int(input())
count = 0
f1 = sushu(3)
for i in range(3, num - 1, 2):
    f2 = sushu(i + 2)
    if f1 and f2:
        count += 1
    f1 = f2
print(count)