PAT乙级 1007. 素数对猜想

让我们定义 d_n 为：d_n = p_n+1 - p_n，其中 p_i 是第i个素数。显然有 d₁=1 且对于n>1有 d_n 是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。

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

输入格式：每个测试输入包含1个测试用例，给出正整数N。

输出格式：每个测试用例的输出占一行，不超过N的满足猜想的素数对的个数。

输入样例：

20

输出样例：

4

代码如下（python）：

__author__ = 'lywade'

def judge_prime(n):
    if n <= 1:
        return False

    i = 2
    while i * i <= n:
        if n % i ==0:
            return False
        i += 1

    return True

num = input()
prime_list = []
count = 0
for i in range(2, num+1):
    if judge_prime(i):
        prime_list.append(i)

num_prime = len(prime_list)
for j in range(num_prime-1):
    if prime_list[j+1] - prime_list[j] == 2:
        count += 1
print count