15
\$\begingroup\$

I did a Goldbach's Conjecture exercise and got it to work. It's pretty slow though and I was wondering how I could optimize it.

number = int(input("Enter your number >> "))
print("\nCalculating...")
if number % 2 == 0: #Only even numbers
    primes = primenums(number) #returns all prime numbers <= input number
    addend1 = primes[0]
    addend2 = primes[0]

    while addend1 + addend2 != number:
        if primes[-1] == addend2:
            addend2 = primes[primes.index(addend1) + 1]
            addend1 = primes[primes.index(addend1) + 1]
        else:
            addend2 = primes[primes.index(addend2) + 1]

Right now, up to 10,000 the algorithm is pretty fast, but at 100,000 it takes about 3 seconds to finish. Is that just how it is or could I make it faster?

\$\endgroup\$
9
  • 8
    \$\begingroup\$ Have you profiled this? How is primenums implemented? \$\endgroup\$
    – Graipher
    Commented Feb 13, 2019 at 8:23
  • 2
    \$\begingroup\$ The first problem is that you're trying to get speed out of arithmetic in Python; that's a losing proposition. Python has many strengths, and speed of basic arithmetic is not one of them. Running your code in an optimizing Python runtime will help considerably, but if you have heavy-duty computations to perform, picking a language that compiles arithmetic down closer to the metal will help enormously. \$\endgroup\$ Commented Feb 13, 2019 at 17:23
  • 7
    \$\begingroup\$ @EricLippert Don't worry about language/compiler/interpreter optimization before you do mathematical/complexity optimization first. Switching languages can only improve performance by a small factor (1-10). Changing the algorithm can give speedups of 1-infinity. \$\endgroup\$
    – Vortico
    Commented Feb 13, 2019 at 18:10
  • 2
    \$\begingroup\$ @Vortico are you seriously lecturing Eric Lippert on performance gains and optimization? \$\endgroup\$ Commented Feb 13, 2019 at 19:57
  • 10
    \$\begingroup\$ @MikeTheLiar I know he knows, but someone reading his comment may be misled. It's bad advice to recommend to a programming beginner to change the tool he's learning before even pointing out the issue with his algorithm. It's discouraging, doesn't solve the problem at hand (because maybe his goal is to simply learn Python, and remember this is a code review community), and doesn't teach how to think on one's own. It's not "The first problem" with his code as he said. It's the 26th or maybe 126th problem he should worry about while playing around with Goldbach's conjecture. \$\endgroup\$
    – Vortico
    Commented Feb 13, 2019 at 20:31

4 Answers 4

25
\$\begingroup\$

One thing that makes your code slow is your repeated calls to primes.index. Each of these calls needs to linearly scan primes until it finds the value you are looking for, making this \$\mathcal{O}(n)\$. This is especially bad in the first if case since you do this twice with exactly the same argument. Instead just keep the index around in addition to the number:

def goldbach_keep_index(number):
    if number % 2 == 1: #Only even numbers
        raise ValueError("Goldbach conjecture is only defined for even numbers")
    primes = primenums(number) #returns all prime numbers <= input number
    i, j = 0, 0
    addend1 = primes[i]
    addend2 = primes[j]

    while addend1 + addend2 != number:
        if addend2 == primes[-1]:
            i = j = i + 1
            addend2 = primes[i]
            addend1 = primes[i]
        else:
            j += 1
            addend2 = primes[j]
    return addend1, addend2

Note that I made this a function, so you can reuse it.

Your main calling code could look like this:

if __name__ == "__main__":
    number = int(input("Enter your number >> "))
    p1, p2 = goldbach_keep_index(number)
    print(f"{p1} + {p2} = {number}")

However, what you are really doing is taking all combinations of prime numbers, without repeating yourself. So just use itertools.combinations_with_replacement:

from itertools import combinations_with_replacement

def goldbach_combinations(number):
    if number % 2 == 0:
        raise ValueError("Goldbach conjecture is only defined for even numbers")
    primes = primenums(number)
    for addend1, addend2 in combinations_with_replacement(primes, 2):
        if addend1 + addend2 == number:
            return addend1, addend2
    raise Exception(f"Found a counter-example to the Goldbach conjecture: {number}")

In this case primes does not even need to be a list, it can just be a generator.


If you instead make primes a set, you can easily implement the idea suggested by @Josiah in their answer by just checking if number - p is in primes:

def goldbach_set(number):
    if number % 2 == 0: #Only even numbers
        raise ValueError("Goldbach conjecture is only defined for even numbers")
    primes = set(primenums(number)) #returns all prime numbers <= input number
    for p in primes:
        k = number - p
        if k in primes:
            return p, k
    raise Exception(f"Found a counter-example to the Goldbach conjecture: {number}")

And now some timing comparisons, where goldbach is your code put into a function:

def goldbach(number):
    if number % 2 == 0: #Only even numbers
        raise ValueError("Goldbach conjecture is only defined for even numbers")
    primes = primenums(number) #returns all prime numbers <= input number
    addend1 = primes[0]
    addend2 = primes[0]

    while addend1 + addend2 != number:
        if primes[-1] == addend2:
            addend2 = primes[primes.index(addend1) + 1]
            addend1 = primes[primes.index(addend1) + 1]
        else:
            addend2 = primes[primes.index(addend2) + 1]
    return addend1, addend2

And primenums is a simple sieve:

def prime_sieve(limit):
    prime = [True] * limit
    prime[0] = prime[1] = False

    for i, is_prime in enumerate(prime):
        if is_prime:
            yield i
            for n in range(i * i, limit, i):
                prime[n] = False

def primenums(limit):
    return list(prime_sieve(limit))

enter image description here

And when pulling the generation of the primes outside of the function (and calculating them up to the largest number in the plot):

enter image description here

Of course the first three functions are vastly slower than the set because they need to go through more combinations now. However, all functions are then constant time, so the increase in time comes solely from the fact that you have to consider more primes.

\$\endgroup\$
8
  • \$\begingroup\$ I'm intrigued by the last graph. There is an obvious oscillating pattern where there's a sudden drop in run time even when the size of the input increases. I wonder what's going in there \$\endgroup\$
    – DeepSpace
    Commented Feb 14, 2019 at 12:06
  • 1
    \$\begingroup\$ I wondered that. My guess is fixed probe numbers are used, and some happen to have a larger small addend needed to satisfy the conjecture, so they take more loop iterations. \$\endgroup\$
    – Josiah
    Commented Feb 14, 2019 at 20:35
  • \$\begingroup\$ @Josiah: That is indeed what I did (x = np.logspace(1, 5, dtype=int); x[x % 2 == 1] += 1). And while larger numbers indeed tend to have more pairs of prime numbers that sum to them, there is quite a lot of variation: en.wikipedia.org/wiki/Goldbach%27s_conjecture#/media/… One might be able to argue that this rise is slightly visible above 10^4 in my plot for the first functions, but since the OP's code took too long I could no go up much higher in numbers... \$\endgroup\$
    – Graipher
    Commented Feb 15, 2019 at 8:05
  • \$\begingroup\$ I would also exclude the case n < 4. \$\endgroup\$
    – gnasher729
    Commented Jan 29, 2022 at 17:29
  • 1
    \$\begingroup\$ In "the index-manipulating variant", I find addend2 = primes[i] counter-intuitive. Just put addend2 = primes[j] after the if-else. \$\endgroup\$
    – greybeard
    Commented Jun 28, 2023 at 7:32
11
\$\begingroup\$

As suggested by Graipher in the comment, you're probably spending a lot of time generating a list of primes. That would be worth profiling, but I can't speculate as to how it might be improved without at least seeing the code.

There are two things that stand out to me as very clear performance sinks.

The first, and easiest to change, is index. If you call index on a list, python has to check every element in the list one at a time to see whether it's the thing you're after. If you have millions of primes, every one of those index calls is secretly a hidden loop that runs millions of times. The good news is you can resolve this easily enough: just keep another variable which stores where in the list of primes you're up to. Then you don't have to search for your place again every time you go round the loop, because you have a bookmark to it.

Second, and slightly more subtle, is the way that this is looping. You are looking for two numbers that add up to a target. If your target is 100, you might first say "what if it's 3?" and then try all primes with 3. Then you say "what if it's 5?" and try all the primes with 5. And then you do the same with 7, and so on. This means your trying all the primes with all the primes.

Instead, if you are checking whether the first prime is 3, you know that the second prime has to be 97. So all you have to do is check whether 97 is prime. If so, it's a match, if not, you can move on to checking 5 and 95, and then 7 and 93, and so on. In this way your algorithm checks each prime against one other number instead of each prime against all the others, and should be massively faster.

\$\endgroup\$
4
  • \$\begingroup\$ Your second suggestion is great indeed. I will implement it that way! Thanks \$\endgroup\$
    – Kevin
    Commented Feb 13, 2019 at 14:03
  • \$\begingroup\$ You can make this even more efficient by keeping track of "near misses" to your target which eliminate other integers. Suppose you were testing 102. You start by looking for 3 + some prime. 99 isn't a prime, but the nearest prime >= 99 is 101, so if you organize the search carefully, you can remember that you just found a prime pair for 104 = 3 + 99 while looking for a pair for 102, and the next even number you need to test is not 104 but 106 (unless you already found a pair for that as well. of course). \$\endgroup\$
    – alephzero
    Commented Feb 13, 2019 at 17:10
  • \$\begingroup\$ I would default to using a set as in @Graipher's implementation, which is very fast at discovering whether something is present, but does not tell you what the nearest miss was. \$\endgroup\$
    – Josiah
    Commented Feb 13, 2019 at 20:18
  • \$\begingroup\$ If I had to use a regular list, and if the list was sorted, then there is indeed a clever trick available. The idea is that as the small prime being tested grows, the number it needs to be added to shrinks. That means that you can keep two bookmark variables pointing at the list of primes, one at the start and moving forwards, and one at the end and moving backwards. As long as sum is too small, you move the first bookmark along. If the sum is too big, you bring the second bookmark backwards. This avoids having to scan the whole list each time. However, it's quite fiddly to get right. \$\endgroup\$
    – Josiah
    Commented Feb 13, 2019 at 20:25
4
\$\begingroup\$

Let's take n = 1,000,000,000 for example and see what your code does.

It calculates all primes from 1 to 1 billion. That takes a while, but it gives you an array of all primes in sorted order.

It then calculates 2 + 2, 2 + 3, 2 + 5, 2 + 7, 2 + 999,999,xxx to check if one of these numbers equals 1,000,000,000. But obviously when addend1 = 2, addend2 has to be 999,999,998 to add to one billion, so you are checking tens of millions of sums unnecessarily.

It then calculates 3 + 3, 3 + 5, 3 + 7 etc., and again, addend2 would have to be 999,999,997 million. And then again for addend1 = 5, 7, 11 etc. So you are checking a huge number of sums needlessly.

Start with addend1 = first prime, addend2 = last prime in your array. If their sum is too small, replace addend1 with the next prime to make the sum larger by the smallest possible amound. If their sum is too large, replace addend2 with the previous prime. If the sum is right, you have found a solution. And once you reached addend1 > addend2, you know there is no solution. This will be very quick, since usually you don't need to check too many values for addend1.

That makes the search a lot quicker, but doesn't help with the sieve trying to find all the primes. You usually don't need all the primes, only a very small number. In the example with n = 1,000,000,000 you probably find two primes adding up to a billion with addend1 ≤ 1000 and addend2 ≥ 999,999,000.

So here's what you do: To find all primes say in the range 999,999,000 to 1,000,000,000, you need to check if these numbers are divisible by any prime up to the square root of 1,000,000,000 which is a bit less than 32,000. So first you use a sieve to find all numbers up to 32,000. Then you create a sieve that finds all primes from 999,999,000 to 1,000,000,000. You run your search algorithm until it tries to examine primes that are not in this range. This likely doesn't happen, but if it happens, you create another sieve for the numbers 999,998,000 to 999,999,000 and so on. So instead of maybe 50 million primes, you look only for maybe 50 or 100. Again, that makes it a lot faster.

\$\endgroup\$
3
  • \$\begingroup\$ I didnt learn about sieves yet, gotta look into it \$\endgroup\$
    – Kevin
    Commented Feb 14, 2019 at 5:52
  • \$\begingroup\$ It might be worth testing the primality of the larger number as you hit it, rather than sieving a whole range. \$\endgroup\$
    – Josiah
    Commented Feb 14, 2019 at 7:54
  • \$\begingroup\$ Start at the big end and exponentially search for the complement. \$\endgroup\$
    – greybeard
    Commented Jun 28, 2023 at 7:33
2
\$\begingroup\$

This will be much faster. (less than a second to check 44 trillion) It does fewer prime checks, usually less than 100, and does not check numbers that are even for prime. Building a sieve of 700k primes for goldbach(12,000,000) takes forever. This code makes 7 IsPrime() calls for 12 million.
With native Python Prime code it takes about .001 second to find answer for 1 Billion making 42 calls to IsPrime().

Three major differences from other IsPrime(n-p) programs:

Checks Goldbach prime 3 then uses 6k ± 1 for the rest. So 1/3 of the Goldbach checks vs every odd number > 3.

More efficient IsPrime() tests using every 6th odd number using 6k ± 1 as well.

Can use GMPY2 to check for prime. (takes less than .001 second for 25 digit number.

Solutions using a sieve get n*k slower as the number size (k) grows. 10,000,000 takes 2.5 seconds 100,000,000 takes 25 seconds and so on until memory is exhausted building the sieve. The call prime_check only as needed grows linearly (k) as k grows: 1,000,000,000,000 takes 0.037 seconds while 100,000,000,000,000 takes 0.450 seconds.

With GMPY2:

$ python3 goldbach.py 12345678912345678
This is a test of Goldbach's Conjecture that for all even integers
greater than 2 there are two primes that add up to that even number.


61 + 12,345,678,912,345,617 = 12,345,678,912,345,678

         42 function calls in 0.000 seconds

   Ordered by: standard name

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    0.000    0.000    0.000    0.000 <string>:1(<module>)
        1    0.000    0.000    0.000    0.000 goldbach.py:16(goldbach)
        1    0.000    0.000    0.000    0.000 {built-in method builtins.exec}
        1    0.000    0.000    0.000    0.000 {built-in method builtins.print}
       37    0.000    0.000    0.000    0.000 {built-in method gmpy2.gmpy2.is_prime}
        1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}


$ python3 goldbach.py 4444444444444444444444
This is a test of Goldbach's Conjecture that for all even integers
greater than 2 there are two primes that add up to that even number.


257 + 4,444,444,444,444,444,444,187 = 4,444,444,444,444,444,444,444

         144 function calls in 0.000 seconds

   Ordered by: standard name

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    0.000    0.000    0.000    0.000 <string>:1(<module>)
        1    0.000    0.000    0.000    0.000 goldbach.py:16(goldbach)
        1    0.000    0.000    0.000    0.000 {built-in method builtins.exec}
        1    0.000    0.000    0.000    0.000 {built-in method builtins.print}
      139    0.000    0.000    0.000    0.000 {built-in method gmpy2.gmpy2.is_prime}
        1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}

With my Home Grown IsPrime:

    $ python3 goldbach.py 12345678912345678
This is a test of Goldbach's Conjecture that for all even integers
greater than 2 there are two primes that add up to that even number.


61 + 12,345,678,912,345,617 = 12,345,678,912,345,678

         42 function calls in 3.869 seconds

   Ordered by: standard name

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    0.000    0.000    3.869    3.869 <string>:1(<module>)
        1    0.000    0.000    3.869    3.869 goldbach.py:16(goldbach)
       37    3.869    0.105    3.869    0.105 goldbach.py:6(IsPrime)
        1    0.000    0.000    3.869    3.869 {built-in method builtins.exec}
        1    0.000    0.000    0.000    0.000 {built-in method builtins.print}
        1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}

    $ python3 goldbach.py 1000000000
This is a test of Goldbach's Conjecture that for all even integers
greater than 2 there are two primes that add up to that even number.


71 + 999,999,929 = 1,000,000,000

         47 function calls in 0.001 seconds

   Ordered by: standard name

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    0.000    0.000    0.001    0.001 <string>:1(<module>)
        1    0.000    0.000    0.001    0.001 goldbach.py:16(goldbach)
       42    0.001    0.000    0.001    0.000 goldbach.py:6(IsPrime)
        1    0.000    0.000    0.001    0.001 {built-in method builtins.exec}
        1    0.000    0.000    0.000    0.000 {built-in method builtins.print}
        1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}

Listing:

# Goldbach's Conjecture tester. 
from gmpy2 import is_prime
import sys
import cProfile
# or use home grown IsPrime
def IsPrime(n): 
   if (n<=1): return False
   if (n <= 3): return True 
   if (n % 2 == 0 or n % 3 == 0): 
          return False 
   # use  6k ± 1 for the rest
   stop = int(n**.5)+1
   for i  in range(5, stop,  6): 
       if (n % i == 0 or n % (i + 2) == 0): 
          return False 
   return True 

def goldbach(number):
    if number == 4:
        print("\n2 + 2 = 4\n")
        return
    elif IsPrime(number - 3):
        print(f"\n3 + {number-3:,} = {number}\n")
        return  
    else:
        for p in range(5, number, 6): # just odds after 3 
            if IsPrime(p) and IsPrime(number-p): 
                print(f"\n{p:,} + {number-p:,} = {N:,}\n")
                return
            elif IsPrime(p+2) and IsPrime(number-(p+2)):
                print(f"\n{p+2:,} + {number-(p+2):,} = {N:,}\n")
                return
        raise Exception(f"Found a counter-example to the Goldbach conjecture: {number}")

if __name__=="__main__":
    N = 1
    args = len(sys.argv)
    if args > 1:
        N = int(sys.argv[1])
    print("This is a test of Goldbach's Conjecture that for all even integers")
    print("greater than 2 there are two primes that add up to that even number.\n")
    while (N < 3 or N%2):
        N = int(input("Please enter an even number > 3 to check with Goldbach's Conjecture> "))
    cProfile.run('goldbach(N)')
\$\endgroup\$
3
  • \$\begingroup\$ This replacement code contains very little in the way of review of the question. And it seems to just be a rehash of a small part of other answers (namely to test n-p for primality). \$\endgroup\$ Commented Jun 28, 2023 at 8:24
  • 1
    \$\begingroup\$ @TobySpeight Take another look at this answer. \$\endgroup\$
    – pacmaninbw
    Commented Jul 22, 2023 at 12:05
  • \$\begingroup\$ @TobySpeight I tested all of the other code in the article. My code is at least an order of magnitude faster. It also checks for prime. None of the sieves can check a 30 digit number at all. The original question was: "It's pretty slow though and I was wondering how I could optimize it." I think my code answers that question. \$\endgroup\$ Commented Sep 22, 2023 at 14:14

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.