diff --git a/src/Traversal.java b/src/Traversal.java index 8da0f79..f922ed4 100644 --- a/src/Traversal.java +++ b/src/Traversal.java @@ -1,5 +1,188 @@ +import java.util.HashMap; +import java.util.LinkedList; +import java.util.Queue; +import java.util.List; +import java.util.Map; +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("are", null, null); + stringRoot.left.right = new TreeNode("hello", null, null); + + stringRoot.right = new TreeNode("I", null, null); + stringRoot.right.left = new TreeNode("am", null, null); + stringRoot.right.right = new TreeNode("am", 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; + } + + levelOrder(root); + + //preOrderIter(megaRoot); + + // System.out.println("Preorder recursive"); + // preorder(root); + // System.out.println("Preorder iterative"); + // preOrderIter(root); + + TreeNode newRoot = new TreeNode(3, new TreeNode(9, null, new TreeNode(4, null, null)), null); + //newRoot.left = new TreeNode(9, null, null); + //newRoot.left.right = new TreeNode(4, null, null); + newRoot.right = new TreeNode(4, null, null); + newRoot.right.left = new TreeNode(9, null, null); + newRoot.right.right = new TreeNode(6, null, null); + newRoot.right.right.left = new TreeNode(4, null, null); + + // { "hello": 1, "how": 2, "you": 1 } + List wordList = List.of("hello", "how", "you", "how"); + //System.out.println(listToMap(wordList)); + + //preorder(root); + //postorder(root); + //inorder(stringRoot); + + //printGreater(root, 4); + //System.out.println(countNodes(root)); + + //System.out.println(toMap(stringRoot)); + } + + // Print a tree rooted at the given node in pre-order + public static void preorder(TreeNode node) { + if (node == null) return; // if null, return + System.out.println(node.value); // print value + preorder(node.left); // traverse left + preorder(node.right); // traverse right + } + + // depth first search + public static void preOrderIter(TreeNode node) { + Stack> stack = new Stack<>(); + stack.push(node); + + while(!stack.empty()) { + TreeNode current = stack.pop(); + if (current == null) { + continue; + } + System.out.println(current.value); + stack.push(current.right); + stack.push(current.left); + } + } + + // breath first search (level order) + public static void levelOrder(TreeNode node) { + Queue> queue = new LinkedList<>(); + 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 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 Map listToMap(List list){ + +// for word in list: +// if our word is a key in the map: +// newValue = map.get(word) + 1 +// map.put(word, newValue) +// else: +// map.put(word, 1) + + // "hello", "how", "you", "how" + + Map map = new HashMap<>(); // {"hello": 1, "how": 1, "you": 1} + + int count = 0; + + for (int i = 0; i < list.size(); i++) { + String word = list.get(i); // how + + if (map.containsKey(word)){ + count = map.get(word) + 1; // value of count = 2 + map.put(word, count); + } else { + map.put(word, 1); + } + } + return map; + } + + + public static int countNodes(TreeNode node) { + 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 void printGreater(TreeNode node, int threshold) { + if (node == null) return; + if (node.value > threshold) { + System.out.println(node.value); + } + printGreater(node.left, threshold); + printGreater(node.right, threshold); + } + + // Print a tree rooted at the given node in post-order + public static void postorder(TreeNode node) { + if (node == null) return; + postorder(node.left); + postorder(node.right); + System.out.println(node.value); + } + // Print a tree rooted at the given node in in-order + public static void inorder(TreeNode node) { + if (node == null) return; + inorder(node.left); + System.out.println(node.value); + inorder(node.right); } -} +} \ No newline at end of file diff --git a/src/TreeNode.java b/src/TreeNode.java index acd9639..71bb3ca 100644 --- a/src/TreeNode.java +++ b/src/TreeNode.java @@ -1,4 +1,11 @@ -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