Pentagonal Number Theorem - Example Program

Example Program

Here is a simple Python program which computes p(n), the number of partitions, using the recurrence resulting from the pentagonal number theorem.

pentagonal = lambda n : n*(3*n-1)/2 def generalised_pentagonal(n): # 0, 1, -1, 2, -2 if n < 0: return 0 if n%2 == 0: return pentagonal(n/2+1) else: return pentagonal(-(n/2+1)) pt = for n in range (1, 1000+1): r = 0 f = -1 i = 0 while 1: k = generalised_pentagonal(i) if k > n: break if i%2==0: f = -f r += f*pt i += 1 pt.append(r) print pt

Read more about this topic:  Pentagonal Number Theorem

Famous quotes containing the word program:

    “[T]he minister preached a sermon on Jonah and the whale, at the end of which an old chief arose and declared, “We have heard several of the white people talk and lie; we know they will lie, but this is the biggest lie we ever heard.””
    —Administration in the State of Miss, U.S. public relief program (1935-1943)

    “Music is so much a part of their daily lives that if an Indian visits another reservation one of the first questions asked on his return is: “What new songs did you learn?””
    —Federal Writers’ Project Of The Wor, U.S. public relief program (1935-1943)