aboutsummaryrefslogtreecommitdiff
path: root/challenge-168/lubos-kolouch/python/ch-1.py
blob: 8c8790c5dc9a23b064d98b57b88169e6c90a1484 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
""" Challenge 168 Task 1 LK Python """

from sympy import isprime


def generate_perrin_primes(n: int) -> list:
    """
    Generate all primes up to n using the Perrin sequence.
    """
    perrin_sequence = [3, 0, 2]

    perrin_primes: dict = {}

    while len(perrin_primes.keys()) < n:

        next_number = perrin_sequence[-3] + perrin_sequence[-2]
        perrin_sequence.append(next_number)

        if isprime(next_number):
            perrin_primes[next_number] = True

    return sorted(perrin_primes.keys())


assert generate_perrin_primes(13) == [
    2,
    3,
    5,
    7,
    17,
    29,
    277,
    367,
    853,
    14197,
    43721,
    1442968193,
    792606555396977,
]