mirror of
https://github.com/TheAlgorithms/JavaScript.git
synced 2026-03-13 15:21:15 +08:00
* 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:
@@ -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 }
|
||||
|
||||
Reference in New Issue
Block a user