1 files changed, 18 insertions, 15 deletions
diff --git a/final/connectedness/dfs.py b/final/connectedness/dfs.py
index 609ddf6..00e14ee 100644
--- a/final/connectedness/dfs.py
+++ b/final/connectedness/dfs.py
@@ -15,19 +15,22 @@ def is_graph_connected(G):
         return len(VISITED) == len(G.nodes)
     return dfs_connectedness(0)
 
-x = np.zeros(400)
-y = np.zeros(400)
-for n in range(1, 400):
-    c = 0
-    for _ in range(5):
-        graph = nx.random_geometric_graph(n, 0.125)
-        x[n] = len(graph.nodes) + len(graph.edges)
-        start = time.time()
-        is_graph_connected(graph)
-        c += time.time() - start
-    c /= 5
-    y[n] = c
-    print(n)
+def main():
+    x = np.zeros(400)
+    y = np.zeros(400)
+    for n in range(1, 400):
+        c = 0
+        for _ in range(5):
+            graph = nx.random_geometric_graph(n, 0.125)
+            x[n] = len(graph.nodes) + len(graph.edges)
+            start = time.time()
+            is_graph_connected(graph)
+            c += time.time() - start
+        c /= 5
+        y[n] = c
 
-plt.scatter(x, y)
-plt.show()
+    plt.scatter(x, y)
+    plt.show()
+
+if __name__ == "__main__":
+    main()