Euler 154

euler154.py

import cProfile
import collections

def pascal(first, length):
    row = []
    a = first
    for x in range(1, length):
        # x and y form the ratio between consecutive elements
        b = length - x
        a = a * b // x # // is the integer division function
        row.append(a)
    return row

def pascalTriangleRow(n):
    return pascal(1, n+1)

def pascalPyramidLayer(n):
    count = 0

    def checkRow(a, row_length):
        for x in range(1, row_length):
            # x and y form the ratio between consecutive elements
            y = row_length - x
            a = a * y // x
            if a > 200000:
                ++count

    length = n + 1
    for x in pascalTriangleRow(n):
        checkRow(x, length)
        --length

    return count

def main():
    print(pascalPyramidLayer(3000))

cProfile.run("main()")