Add catalan_numbers.py

Author: Kommandat

dynamic_programming/catalan_numbers.py

@@ -0,0 +1,94 @@
+"""
+Print all the Catalan numbers from 0 to n, n being the user input.
+
+ * The Catalan numbers are a sequence of positive integers that
+ * appear in many counting problems in combinatorics [1]. Such
+ * problems include counting [2]:
+ * - The number of Dyck words of length 2n
+ * - The number well-formed expressions with n pairs of parentheses
+ *   (e.g., `()()` is valid but `())(` is not)
+ * - The number of different ways n + 1 factors can be completely
+ *   parenthesized (e.g., for n = 2, C(n) = 2 and (ab)c and a(bc)
+ *   are the two valid ways to parenthesize.
+ * - The number of full binary trees with n + 1 leaves
+
+ * A Catalan number satisfies the following recurrence relation
+ * which we will use in this algorithm [1].
+ * C(0) = C(1) = 1
+ * C(n) = sum(C(i).C(n-i-1)), from i = 0 to n-1
+
+ * In addition, the n-th Catalan number can be calculated using
+ * the closed form formula below [1]:
+ * C(n) = (1 / (n + 1)) * (2n choose n)
+
+ * Sources:
+ *  [1] https://brilliant.org/wiki/catalan-numbers/
+ *  [2] https://en.wikipedia.org/wiki/Catalan_number
+"""
+
+
+class Catalan:
+    def __init__(self, N: int):
+        self.catalan_numbers = [0] * (N + 1)
+        if N:
+            N = int(N)
+            # Base case: C(0) = C(1) = 1
+            self.catalan_numbers[0] = 1
+            self.catalan_numbers[1] = 1
+
+            # Recurrence relation: C(i) = sum(C(j).C(i-j-1)), from j = 0 to i
+            for i in range(2, N + 1):
+                for j in range(i):
+                    self.catalan_numbers[i] += (
+                        self.catalan_numbers[j] * self.catalan_numbers[i - j - 1]
+                    )
+        elif N == 0:
+            self.catalan_numbers[0] = 1
+        print(self.catalan_numbers)
+
+    def get(self, sequence_no=None):
+        """
+        >>> Catalan(5).get(3)
+        [1, 1, 2, 5, 14, 42]
+        [1, 1, 2, 5]
+        >>> Catalan(5).get(6)
+        [1, 1, 2, 5, 14, 42]
+        Out of bound.
+        >>> Catalan(5).get(-1)
+        [1, 1, 2, 5, 14, 42]
+        []
+        """
+        if sequence_no is not None:
+            if sequence_no < len(self.catalan_numbers):
+                return print(self.catalan_numbers[: sequence_no + 1])
+            else:
+                print("Out of bound.")
+        else:
+            print("Please specify a value")
+
+
+if __name__ == "__main__":
+    print("\n********* Catalan Numbers Using Dynamic Programming ************\n")
+    print("\n Enter the upper limit for the Catalan number sequence: ", end="")
+    try:
+        N = int(input().strip())
+        cat = Catalan(N)
+        print(
+            "\n********* Enter different values to get the corresponding Catalan "
+            "Number sequence, enter any negative number to exit. ************\n"
+        )
+        while True:
+            try:
+                i = int(input("Enter value: ").strip())
+                if i < 0:
+                    print("\n********* Good Bye!! ************\n")
+                    break
+                cat.get(i)
+            except NameError:
+                print("\nInvalid input, please try again.")
+    except NameError:
+        print("\n********* Invalid input, good bye!! ************\n")
+
+    import doctest
+
+    doctest.testmod()