fixes: #1280 and test: added tests for DFS algorithm (#1303)

* fix: fixed error in DepthFirstSearch algorithm and test: added tests for DepthFirstSearch algorithm.

* changed traverseDFS function parameters in DepthFirstSearch.js file
This commit is contained in:
Ranganathan7
2023-03-02 00:03:24 +05:30
committed by GitHub
parent 566d9103cd
commit fc06690672
2 changed files with 62 additions and 19 deletions

View File

@@ -4,35 +4,32 @@
* DFS Algorithm for traversing or searching graph data structures.
*/
function traverseDFS (root) {
const stack = [root]
// traverses a give tree from specified root's value
function traverseDFS (tree, rootValue) {
const stack = []
const res = []
stack.push(searchDFS(tree, rootValue))
// if root is not present in the tree, returning empty array
if (!stack[0]) return res
while (stack.length) {
const curr = stack.pop()
res.push(curr.key)
if (curr.right) {
stack.push(curr.right)
}
res.push(curr.value)
if (curr.left) {
stack.push(curr.left)
stack.push(tree[curr.left])
}
if (curr.right) {
stack.push(tree[curr.right])
}
}
return res.reverse()
}
function searchDFS (tree, value) {
const stack = []
stack.push(tree[0])
while (stack.length !== 0) {
for (let i = 0; i < stack.length; i++) {
const node = stack.pop()
if (node.value === value) {
return node
}
@@ -59,11 +56,10 @@ const tree = [
{ value: 10, left: null, right: null },
{ value: 1, left: null, right: null }
]
searchDFS(tree, 9)
searchDFS(tree, 10)
traverseDFS(6)
searchDFS(tree, 9) // { value: 9, left: 7, right: 8 }
searchDFS(tree, 200) // null
traverseDFS(tree, 6) // [ 1, 2, 3, 4, 5, 8, 10, 9, 7, 6 ]
traverseDFS(tree, 200) // []
// 6
// / \
@@ -74,3 +70,4 @@ traverseDFS(6)
// 2 8 10
// /
// 1
export { searchDFS, traverseDFS }