本文用遞歸算法實現二叉樹的前序、中序和后序遍歷,提供Java版的基本模板,在模板上稍作修改,即可解決LeetCode144. Binary Tree Preorder Traversal(二叉樹前序遍歷),94. Binary Tree Inorder Traversal(二叉樹中序遍歷),145. Binary Tree Postorder Traversal(二叉樹后序遍歷)。
基本概念
二叉樹的遍歷是根據訪問結點操作發生位置命名:
前序:訪問根結點的操作發生在遍歷其左右子樹之前。
中序:訪問根結點的操作發生在遍歷其左右子樹之中。
后序:訪問根結點的操作發生在遍歷其左右子樹之后。
模板准備
模板比較簡單,廢話不說了,直接上代碼。
二叉樹節點定義:
//Definition for a binary tree node. public class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int x) { val = x; } }
前序遍歷:
public void preOrder(TreeNode node){ if(node == null){ return; } System.out.println(node.val); preOrder(node.left); preOrder(node.right); }
中序遍歷:
public void inOrder(TreeNode node){ if(node == null){ return; } inOrder(node.left); System.out.println(node.val); inOrder(node.right); }
后序遍歷:
public void postOrder(TreeNode node){ if(node == null){ return; } postOrder(node.left); postOrder(node.right); System.out.println(node.val); }
LeetCode題解
LeetCode上面關於二叉樹遍歷的問題有:
| # | Title |
| 144 | Binary Tree Preorder Traversal |
| 94 | Binary Tree Inorder Traversal |
| 145 | Binary Tree Postorder Traversal |
解答以上問題,只需要在遍歷模板上稍作修改即可:題目要求返回類型是List,所以只需要在訪問元素的時候把元素添加到結果列表中。
144. Binary Tree Preorder Traversal
class Solution { public List<Integer> preorderTraversal(TreeNode root) { List<Integer> ret =new ArrayList<Integer>(); preOrder(root,ret); return ret; } public void preOrder(TreeNode node,List<Integer> ret){ if(node == null){ return ; } ret.add(node.val); preOrder(node.left,ret); preOrder(node.right,ret); } }
94. Binary Tree Inorder Traversal
class Solution { public List<Integer> inorderTraversal(TreeNode root) { List<Integer> ret = new ArrayList<Integer>(); inOrder(root,ret); return ret; } public void inOrder(TreeNode node,List<Integer> ret){ if(node == null){ return ; } inOrder(node.left,ret); ret.add(node.val); inOrder(node.right,ret); } }
145. Binary Tree Postorder Traversal
class Solution { public List<Integer> postorderTraversal(TreeNode root) { List<Integer> ret =new ArrayList<Integer>(); postOrder(root,ret); return ret; } public void postOrder(TreeNode node,List<Integer> ret){ if(node == null){ return ; } postOrder(node.left,ret); postOrder(node.right,ret); ret.add(node.val); } }
總結
二叉樹是天然的遞歸結構,利用此遞歸結構,可以巧妙地解決許多看似復雜的二叉樹問題。
