1 package Tree; 2 3 import java.util.ArrayDeque; 4 import java.util.Queue; 5 import java.util.Stack; 6 7 /** 8 * Created by lenovo on 2017/9/6. 9 */ 10 public class BinaryTree { 11 /* 12 * 前序遍歷,遞歸實現 13 * */ 14 public void PreOrder(TreeNode node) { 15 if (node != null) { 16 System.out.print(node.val); 17 PreOrder(node.left); 18 PreOrder(node.right); 19 } 20 } 21 22 /* 23 * 前序遍歷,非遞歸實現 24 * 1,先入棧根節點,輸出根節點val值,再先后入棧其右節點、左結點; 25 * 2,出棧左節點,輸出其val值,再入棧該左節點的右節點、左節點;直到遍歷完該左節點所在子樹。 26 * 3,再出棧右節點,輸出其val值,再入棧該右節點的右節點、左節點;直到遍歷完該右節點所在子樹。 27 * */ 28 public void PreOrder1(TreeNode root) { 29 Stack<TreeNode> stack = new Stack<>(); 30 if (root != null) { 31 stack.push(root); 32 } 33 while (!stack.empty()) { 34 TreeNode node = stack.pop(); 35 System.out.print(node.val); 36 //右結點先入棧,左結點后入棧 37 if (node.right != null) stack.push(node.right); 38 if (node.left != null) stack.push(node.left); 39 } 40 } 41 42 /* 43 * 中序遍歷,遞歸實現 44 * */ 45 public void InOrder(TreeNode node) { 46 if (node != null) { 47 InOrder(node.left); 48 System.out.print(node.val); 49 InOrder(node.right); 50 } 51 } 52 53 /* 54 * 中序遍歷,非遞歸實現 55 * 1,首先從根節點出發一路向左,入棧所有的左節點; 56 * 2,出棧一個節點,輸出該節點val值,查詢該節點是否存在右節點, 57 * 若存在則從該右節點出發一路向左入棧該右節點所在子樹所有的左節點; 58 * 3,若不存在右節點,則出棧下一個節點,輸出節點val值,同步驟2操作; 59 * 4,直到節點為null,且棧為空。 60 * */ 61 public void InOrder1(TreeNode root) { 62 Stack<TreeNode> stack = new Stack<>(); 63 while (root != null || !stack.empty()) { 64 while (root != null) { 65 stack.push(root); 66 root = root.left; 67 } 68 if (!stack.empty()) { 69 TreeNode node = stack.pop(); 70 System.out.print(node.val); 71 root = node.right; 72 } 73 } 74 } 75 76 /* 77 * 后序遍歷,遞歸實現 78 * */ 79 public void PostOrder(TreeNode node) { 80 if (node != null) { 81 PostOrder(node.left); 82 PostOrder(node.right); 83 System.out.print(node.val); 84 } 85 } 86 87 /* 88 * 層序遍歷(廣度優先遍歷) 89 * */ 90 public void LayerOrder(TreeNode root) { 91 Queue<TreeNode> queue = new ArrayDeque<>(); 92 if (root != null) queue.offer(root); 93 while (!queue.isEmpty()) { 94 TreeNode node = queue.poll(); 95 System.out.print(node.val); 96 if (node.left != null) queue.offer(node.left); 97 if (node.right != null) queue.offer(node.right); 98 } 99 } 100 }
