From d78be165a85f6af2820ae430e084bf04aa57a37b Mon Sep 17 00:00:00 2001
From: sirjan13 <sirjan13@yahoo.com>
Date: Mon, 2 Oct 2017 12:41:56 +0530
Subject: [PATCH] Added DFS code in python

---
 Python implementation DataStructures/dfs.py | 53 +++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 Python implementation DataStructures/dfs.py

diff --git a/Python implementation DataStructures/dfs.py b/Python implementation DataStructures/dfs.py
new file mode 100644
index 00000000..0dca2e2b
--- /dev/null
+++ b/Python implementation DataStructures/dfs.py	
@@ -0,0 +1,53 @@
+import collections
+
+class Graph:
+    def __init__(self):
+        self.adjList = collections.defaultdict(set)
+
+    def addEdge(self, node1, node2):
+        self.adjList[node1].add(node2)
+        self.adjList[node2].add(node1)
+
+def dfsHelper(current, graph, visited, visitFunc):
+    if (visited[current]):
+        return
+    
+    visited[current] = True;
+    
+    visitFunc(current)
+    
+    for neighbor in graph.adjList[current]:
+        dfsHelper(neighbor, graph, visited, visitFunc)
+    
+
+def dfs(current, graph, visitFunc):
+    visited = collections.defaultdict(bool)
+    dfsHelper(current, graph, visited, visitFunc)
+
+def visitPrint(i):
+    print(i)
+
+# Testing the depth first search implementation
+if __name__ == "__main__":
+
+    # Testing on this tree
+    #      1
+    #     / \
+    #    /   \
+    #   2     3
+    #  / \   / \
+    # 4   5 6   7
+    
+    g = Graph()
+    g.addEdge(1, 2)
+    g.addEdge(1, 3)
+    g.addEdge(2, 4)
+    g.addEdge(2, 5)
+    g.addEdge(3, 6)
+    g.addEdge(3, 7)
+
+    print("Test 1:")
+    dfs(1, g, visitPrint)
+    
+    print("\nTest2:")
+    dfs(2, g, visitPrint)