Created
February 22, 2016 04:59
-
-
Save thmain/e9c99829d4c233b0995f to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import java.util.LinkedList; | |
| import java.util.Queue; | |
| public class LevelOrderTraversal { | |
| public void levelOrderNaiveApproach(Node root){ | |
| int h = height(root); | |
| for(int i=1;i<=h;i++){ | |
| printLevels(root,i); | |
| System.out.println(""); | |
| } | |
| } | |
| public void printLevels(Node root, int h){ | |
| if(root==null) return; | |
| if(h==1) System.out.print(" " + root.data); | |
| else{ | |
| printLevels(root.left,h-1); | |
| printLevels(root.right,h-1); | |
| } | |
| } | |
| public int height(Node root){ | |
| if (root==null) return 0; | |
| return 1 + Math.max(height(root.left),height(root.right)); | |
| } | |
| public void levelOrderQueue(Node root){ | |
| Queue q = new LinkedList(); | |
| int levelNodes =0; | |
| if(root==null) return; | |
| q.add(root); | |
| while(!q.isEmpty()){ | |
| levelNodes = q.size(); | |
| while(levelNodes>0){ | |
| Node n = (Node)q.remove(); | |
| System.out.print(" " + n.data); | |
| if(n.left!=null) q.add(n.left); | |
| if(n.right!=null) q.add(n.right); | |
| levelNodes--; | |
| } | |
| System.out.println(""); | |
| } | |
| } | |
| public static void main (String[] args) throws java.lang.Exception | |
| { | |
| Node root = new Node(5); | |
| root.left = new Node(10); | |
| root.right = new Node(15); | |
| root.left.left = new Node(20); | |
| root.left.right = new Node(25); | |
| root.right.left = new Node(30); | |
| root.right.right = new Node(35); | |
| LevelOrderTraversal i = new LevelOrderTraversal(); | |
| System.out.println(" Output by Naive Approach : "); | |
| i.levelOrderNaiveApproach(root); | |
| System.out.println(" Output by Better Approach : "); | |
| i.levelOrderQueue(root); | |
| } | |
| } | |
| class Node{ | |
| int data; | |
| Node left; | |
| Node right; | |
| public Node(int data){ | |
| this.data = data; | |
| this.left = null; | |
| this.right =null; | |
| } | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Nice implementation !!