add main backtracking/max_colors_example

Author: khanhkhanhlele

backtracking/coloring.py

@@ -111,3 +111,20 @@ def color(graph: list[list[int]], max_colors: int) -> list[int]:
         return colored_vertices
 
     return []
+
+if __name__ == "__main__":
+    # Example graph represented as an adjacency matrix
+    graph_example = [
+        [0, 1, 0, 0, 0],
+        [1, 0, 1, 0, 1],
+        [0, 1, 0, 1, 0],
+        [0, 0, 1, 0, 1],
+        [0, 1, 0, 1, 0],
+    ]
+
+    max_colors_example = 3
+    coloring_result = color(graph_example, max_colors_example)
+    if coloring_result:
+        print(f"Graph can be colored with {max_colors_example} colors: {coloring_result}")
+    else:
+        print(f"Graph cannot be colored with {max_colors_example} colors.")