Create DepthFirstSearch.java
This commit is contained in:
parent
39806e3fe6
commit
6938c409f9
126
src/main/java/com/search/DepthFirstSearch.java
Normal file
126
src/main/java/com/search/DepthFirstSearch.java
Normal file
@ -0,0 +1,126 @@
|
|||||||
|
package src.main.java.com.search;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Searching is faster in sorted structures. Binary search is O(log n).
|
||||||
|
* However, the cost of sorting is O(n · log n).
|
||||||
|
* What to do when adding or removing elements? Sort again? No.
|
||||||
|
* Create efficient data structures to maintain sorted sequences, and search in them
|
||||||
|
* Key example: binary sorted tree, allowing O(log N) insert, remove and lookup.
|
||||||
|
|
||||||
|
In comes, depth-first search
|
||||||
|
* Worst-case performance O(n)
|
||||||
|
* Best-case performance O(1)
|
||||||
|
* Average performance O(n)
|
||||||
|
*
|
||||||
|
* @author abir (https://github.com/abircb)
|
||||||
|
*/
|
||||||
|
|
||||||
|
public class DepthFirstSearch {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Depth-first search method
|
||||||
|
*
|
||||||
|
* @param tree- Binary tree to be searched
|
||||||
|
* @param value- Key being searched for
|
||||||
|
* @return Location of the key
|
||||||
|
*/
|
||||||
|
|
||||||
|
public static <T extends Comparable<T>> T find(T key, BinaryTree<T> tree) {
|
||||||
|
return tree.get(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The BinaryTree class defines the structure of a binary tree
|
||||||
|
* Also contains a static nested class called TreeNode
|
||||||
|
* @param <T>
|
||||||
|
* @author abir
|
||||||
|
*/
|
||||||
|
class BinaryTree<T extends Comparable<T>> {
|
||||||
|
|
||||||
|
private TreeNode<T> root;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author abir
|
||||||
|
* @param <P>
|
||||||
|
* This class defines what a node in a binary tree looks like
|
||||||
|
*/
|
||||||
|
private static class TreeNode<P extends Comparable<P>> {
|
||||||
|
|
||||||
|
P key, value;
|
||||||
|
TreeNode<P> left, right;
|
||||||
|
|
||||||
|
private TreeNode(P key, P value) {
|
||||||
|
this.key = key;
|
||||||
|
this.value = value;
|
||||||
|
this.left = null;
|
||||||
|
this.right = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param node
|
||||||
|
* adds the specified node
|
||||||
|
*/
|
||||||
|
private void add(TreeNode<P> node) {
|
||||||
|
if (node.key.compareTo(this.key) < 0) {
|
||||||
|
if(this.left == null) {
|
||||||
|
this.left = node;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
this.left.add(node);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
else {
|
||||||
|
if(this.right == null) {
|
||||||
|
this.right = node;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
this.right.add(node);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param key
|
||||||
|
* @return the tree node corresponding to the key
|
||||||
|
*/
|
||||||
|
private TreeNode<P> find(P key) {
|
||||||
|
if(key.compareTo(this.key) == 0) return this;
|
||||||
|
|
||||||
|
else if(key.compareTo(this.key) < 0) {
|
||||||
|
if(this.left == null) return null;
|
||||||
|
else return this.left.find(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
else {
|
||||||
|
if(this.right == null) return null;
|
||||||
|
else return this.right.find(key);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public BinaryTree() {
|
||||||
|
this.root = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void add(T key, T value) {
|
||||||
|
TreeNode<T> node = new TreeNode<T>(key, value);
|
||||||
|
if(this.root == null) {
|
||||||
|
this.root = node;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
this.root.add(node);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public T get(T key) {
|
||||||
|
if(this.root == null) return null;
|
||||||
|
|
||||||
|
TreeNode<T> node = this.root.find(key);
|
||||||
|
if(node == null) return null;
|
||||||
|
return node.value;
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user