力扣-树-递归遍历
1.前序遍历(LeetCode144)
- 基本思路:
使用递归完成根-左子树-右子树 - java代码
class Solution {
public List<Integer> preorderTraversal(TreeNode root){
List<Integer> lst=new ArrayList<>();
preorder(lst,root);
return lst;
}
public void preorder(List<Integer> lst,TreeNode root){
if(root==null){
return;
}
lst.add(root.val);
preorder(lst,root.left);
preorder(lst,root.right);
}
}
- python代码
(这里官方给出的解决方法是一个嵌套的“函数”,我尝试了一下耗时会更长一些,下面给出的均为未嵌套方式)
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
"""
:type root: TreeNode
:rtype: List[int]
"""
lst = list()
self.preorder(root,lst)
return lst
def preorder(self, root: TreeNode,lst):
if not root:
return
lst.append(root.val)
self.preorder(root.left,lst)
self.preorder(root.right,lst)
2.中序遍历(LeetCode94)
- 基本思路:
对于递归的解法来说,在完成前序遍历后,中序和后续就很简单了,只需要修改遍历顺序就可以完成 - java代码
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> lst=new ArrayList<>();
inorder(lst,root);
return lst;
}
public void inorder(List<Integer> lst,TreeNode root){
if(root==null){
return;
}
inorder(lst,root.left);
lst.add(root.val);
inorder(lst,root.right);
}
}
- python代码
class Solution:
def inorderTraversal(self, root: TreeNode) -> List[int]:
lst = list()
self.inorder(root, lst)
return lst
def inorder(self, root: TreeNode, lst):
if not root:
return
self.inorder(root.left, lst)
lst.append(root.val)
self.inorder(root.right, lst)
3.后序遍历(LeetCode145)
- java代码
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> lst=new ArrayList<>();
postorder(lst,root);
return lst;
}
public void postorder(List<Integer> lst,TreeNode root){
if(root==null){
return;
}
postorder(lst,root.left);
postorder(lst,root.right);
lst.add(root.val);
}
}
- python代码
class Solution:
def postorderTraversal(self, root: TreeNode) -> List[int]:
lst = list()
self.postorder(root, lst)
return lst
def postorder(self, root: TreeNode, lst):
if not root:
return
self.postorder(root.left, lst)
self.postorder(root.right, lst)
lst.append(root.val)