-
Notifications
You must be signed in to change notification settings - Fork 19.8k
/
Copy pathBreadthFirstSearch.java
71 lines (59 loc) · 2 KB
/
BreadthFirstSearch.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package com.thealgorithms.searches;
import com.thealgorithms.datastructures.Node;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Optional;
import java.util.Queue;
import java.util.Set;
/**
* Breadth-First Search implementation for tree/graph traversal.
* @author caos321
* @co-author @manishraj27
* @see <a href="https://en.wikipedia.org/wiki/Breadth-first_search">Breadth-first search</a>
*/
public class BreadthFirstSearch<T> {
private final List<T> visited = new ArrayList<>();
private final Set<T> visitedSet = new HashSet<>();
/**
* Performs a breadth-first search to find a node with the given value.
*
* @param root The root node to start the search from
* @param value The value to search for
* @return Optional containing the found node, or empty if not found
*/
public Optional<Node<T>> search(final Node<T> root, final T value) {
if (root == null) {
return Optional.empty();
}
visited.add(root.getValue());
visitedSet.add(root.getValue());
if (root.getValue() == value) {
return Optional.of(root);
}
Queue<Node<T>> queue = new ArrayDeque<>(root.getChildren());
while (!queue.isEmpty()) {
final Node<T> current = queue.poll();
T currentValue = current.getValue();
if (visitedSet.contains(currentValue)) {
continue;
}
visited.add(currentValue);
visitedSet.add(currentValue);
if (currentValue == value || (value != null && value.equals(currentValue))) {
return Optional.of(current);
}
queue.addAll(current.getChildren());
}
return Optional.empty();
}
/**
* Returns the list of nodes in the order they were visited.
*
* @return List containing the visited nodes
*/
public List<T> getVisited() {
return visited;
}
}