diff --git a/src/Traversal.java b/src/Traversal.java index 8da0f79..9c2dfeb 100644 --- a/src/Traversal.java +++ b/src/Traversal.java @@ -1,5 +1,204 @@ +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("Root", null, null); + stringRoot.left = new TreeNode("dog", null, null); + stringRoot.left.left = new TreeNode("dog", null, null); + stringRoot.left.right = new TreeNode("cat", null, null); + + stringRoot.right = new TreeNode("house", null, null); + stringRoot.right.left = new TreeNode("hello", null, null); + + stringRoot.right.right = new TreeNode("goodbye", null, null); + stringRoot.right.right.right = new TreeNode("bonito flakes", 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; + + } + + // System.out.println("preorder recursive"); + // preOrder(root); + // postOrder(root); + // inOrder(root); + + + // inOrder(stringRoot); + + // printGreater(root, 4); + + + // System.out.println(countNodes(root)); + + + + // System.out.println(toMap(stringRoot)); + + // preOrder(megaRoot); + // System.out.println("preorder iterator"); + // preOrderIter(root); + // preOrderIter(megaRoot); + levelOrder(root); + } + +// preorder travel using loops/iterations + public static void preOrderIter(TreeNode node){ + Stack> stack = new Stack<>(); + if(node == null) return; + stack.push(node); + + + while(!stack.isEmpty()){ + TreeNode current = stack.pop(); + + if(current == null ) continue; + + System.out.println(current.value); + + stack.push(current.right); + stack.push(current.left); + + } + + } + + + // print a tree rooted at the given node in pre-order + public static void preOrder(TreeNode node){ + + // if null, return + if(node == null){ + return; + } + // print value + System.out.println(node.value); + // taverse left + preOrder(node.left); + // traverse right + preOrder(node.right); + } + + public static void levelOrder(TreeNode node){ + Queue> queue = new LinkedList<>(); + if(node == null) return; + queue.add(node); + + + while(!queue.isEmpty()){ + TreeNode current = queue.poll(); + + if(current == null ) continue; + + System.out.println(current.value); + + queue.add(current.left); + queue.add(current.right); + + + } + + } + + + public static void postOrder(TreeNode node){ + // if null, return + if(node == null){ + return; + } + // taverse left + postOrder(node.left); + // traverse right + postOrder(node.right); + // print value + System.out.println(node.value); + } + + public static void inOrder(TreeNode node){ + // if null, return + if(node == null){ + return; + } + // taverse left + inOrder(node.left); + // print value + System.out.println(node.value); + + // traverse right + inOrder(node.right); + } + + public static void printGreater(TreeNode node, int n){ + if(node == null){ + return; + } + if(node.value > n){ + System.out.println(node.value); + } + printGreater(node.left, n); + printGreater(node.right, n); + }; + + + public static int countNodes(TreeNode node){ + //leftCount = count nodes on left + //rightCount = count nodes on right + //overallCount = leftCount + rightCount + + if(node == null) return 0; + return countNodes(node.left) + countNodes(node.right) + 1; + + // return node == null ? 0 : countNodes(node.left) + countNodes(node.right) + 1; + + // int leftCount = countNodes(node.left); + // int rightCount = countNodes(node.right); + + // int overallCount = leftCount + rightCount + 1; + + // return overallCount; + } + + + 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); + + + } + } diff --git a/src/TreeNode.java b/src/TreeNode.java index acd9639..54bec9b 100644 --- a/src/TreeNode.java +++ b/src/TreeNode.java @@ -1,4 +1,14 @@ -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