diff --git a/src/Traversal.java b/src/Traversal.java index 8da0f79..10ee660 100644 --- a/src/Traversal.java +++ b/src/Traversal.java @@ -1,5 +1,139 @@ +import static org.junit.jupiter.api.DynamicTest.stream; + +import java.util.HashMap; +import java.util.LinkedList; +import java.util.Map; +import java.util.Queue; +import java.util.Stack; + public class Traversal { public static void main(String[] args) { + TreeNode root = new TreeNode(10, null, null); + root.left = new TreeNode(9, null, null); + root.left.left = new TreeNode(5, null, null); + root.left.right = new TreeNode(2, null, null); + + root.right = new TreeNode(15, null, null); + root.right.left = new TreeNode(-3, null, null); + root.right.right = new TreeNode(5, null, null); + root.right.right.right = new TreeNode(22, null, null); + + TreeNode stringRoot = new TreeNode("hello", null, null); + stringRoot.left = new TreeNode("How", null, null); + stringRoot.left.left = new TreeNode("and", null, null); + stringRoot.left.right = new TreeNode("and", null, null); + + stringRoot.right = new TreeNode("I", null, null); + stringRoot.right.left = new TreeNode("am", null, null); + stringRoot.right.right = new TreeNode("so", null, null); + stringRoot.right.right.right = new TreeNode("good", null, null); + + TreeNode megaRoot = new TreeNode(1, null, null); + TreeNode current = megaRoot; + for(int i = 2; i <= 50000; i++) { + current.right = new TreeNode(i, null, null); + current = current.right; + } + + levelOrderIter(root); + + //preOrderIter(megaRoot); + // System.out.println("Preorder Recursive"); + // preOrder(root); + + // System.out.println("Preorder Iterative"); + // preOrderIter(root); + + // preOrder(root); + System.out.println(); + // postOrder(root); + //printGrader(root, 100); + // System.out.println(countNodes(stringRoot)); + + //System.out.println(toMap(stringRoot)); + } + + public static void preOrderIter(TreeNode node) { + Stack> stack = new Stack<>(); + stack.push(node); + while(!stack.isEmpty()) { + TreeNode currentNode = stack.pop(); + + if(currentNode == null) continue; // move onto next iteration, will not stop loop + + System.out.println(currentNode.value); + stack.push(currentNode.right); + stack.push(currentNode.left); + } + } + + public static void levelOrderIter(TreeNode node) { + Queue> queue = new LinkedList<>(); + queue.add(node); + while(!queue.isEmpty()) { + TreeNode currentNode = queue.remove(); + + if(currentNode == null) continue; // move onto next iteration, will not stop loop + + System.out.println(currentNode.value); + queue.add(currentNode.left); + queue.add(currentNode.right); + } + } + + // Print tree rooted at the given node in pre-order. + // ? if you dont need to reuse it, parameterize + public static void preOrder(TreeNode node) { + if (node == null) + return; + System.out.println(node.value); + preOrder(node.left); + preOrder(node.right); + } + + public static Map toMap(TreeNode node) { + Map counts = new HashMap<>(); + toMap(node, counts); + return counts; + } + + private static void toMap(TreeNode node, Map counts) { + if(node == null) return; + // fill up those counts + counts.put(node.value, counts.getOrDefault(node.value, 0) + 1); + toMap(node.left, counts); + toMap(node.right, counts); + } + + public static int countNodes(TreeNode node) { + return node == null ? 0 : 1 + countNodes(node.left) + countNodes(node.right); + } + + public static void printGrader(TreeNode node, int threshold) { + if (node == null) return; + if(node.value > threshold) { + System.out.println(node.value); + } + printGrader(node.left, threshold); + printGrader(node.right, threshold); + } + + + + // type variable if you need to reuse it + public static void postOrder(TreeNode node) { + if (node == null) + return; + postOrder(node.left); + postOrder(node.right); + System.out.println(node.value); + } + public static void inOrder(TreeNode node) { + if (node == null) + return; + inOrder(node.left); + System.out.println(node.value); + inOrder(node.right); } } diff --git a/src/TreeNode.java b/src/TreeNode.java index acd9639..1dbf298 100644 --- a/src/TreeNode.java +++ b/src/TreeNode.java @@ -1,4 +1,12 @@ -public class TreeNode { +public class TreeNode{ + public T value; + public TreeNode left; + public TreeNode right; + public TreeNode(T value, TreeNode left, TreeNode right) { + this.value = value; + this.left = left; + this.right = right; + } } \ No newline at end of file