当前位置:   article > 正文

编程练习【二叉树的中序遍历】_root = [1,null,2,3]

root = [1,null,2,3]

给定一个二叉树的根节点 root ,返回 它的 中序 遍历 。

示例 1:
输入:root = [1,null,2,3]
输出:[1,3,2]
示例 2:

输入:root = []
输出:[]
示例 3:

输入:root = [1]
输出:[1]

题目来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/binary-tree-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:

首先我们要知道关于二叉树的中序遍历的意思是指按照:访问左子树——根节点——右子树的方式遍历这棵树,而在访问左子树或者右子树的时候我们按照同样的方式遍历,直到遍历完整棵树。因此整个遍历过程天然具有递归的性质,我们可以直接用递归函数来模拟这一过程。

方法一:递归
定义一个方法inorder(TreeNode root, List<Integer> result)来实现链表的递归遍历。先遍历节点左子树,再将节点加入集合,后遍历节点右子树,直到节点为空。

  1. public List<Integer> inorderTraversal(TreeNode root) {
  2. List<Integer> result=new ArrayList<Integer>();
  3. inorder(root,result);//递归遍历链表
  4. return result;
  5. }
  6. private void inorder(TreeNode root, List<Integer> result) {
  7. if(root==null) {//当节点为空时,停止遍历
  8. return;
  9. }
  10. inorder(root.left, result);//先遍历左子树
  11. result.add(root.val);//将节点添加进去
  12. inorder(root.right, result);//再遍历右子树
  13. }

 方法二:迭代
和递归方式一样,但创建了一个栈来作为中间存储,节点不为空或者栈不为空时,遍历让节点进栈并遍历左子树,直到节点为空,节点出栈,将节点值添加到集合中,再遍历右子树,直到遍历完整个树。

  1. public List<Integer> inorderTraversal(TreeNode root) {
  2. List<Integer> list=new ArrayList<Integer>();
  3. Stack<TreeNode> stack=new Stack<TreeNode>();//利用一个栈作为零时存储结果
  4. while(root!=null||!stack.isEmpty()) {
  5. while(root!=null) {//当节点不为空时,将节点进栈,且让节点指向左子节点
  6. stack.push(root);
  7. root=root.left;
  8. }
  9. root=stack.pop();//节点出栈
  10. list.add(root.val);//将节点的值添加到集合中
  11. root=root.right;//将节点指向右子节点
  12. }
  13. return list;
  14. }

 方法三:

遍历树,判断当前节点是否还有左孩子,如果有,遍历左子树,找到左子树上最右的节点,记录该节点为predecessor,再判断该节点是否没有右孩子,如果有,则指向其右孩子,再访问左孩子,如果还有右孩子,则说明已经遍历完左子树,将右孩子置空,再将节点加入结果集合;如果当前节点没有左孩子,则将节点加入结果集,再访问右孩子。

  1. public List<Integer> inorderTraversal(TreeNode root) {
  2. List<Integer> res = new ArrayList<Integer>();
  3. TreeNode predecessor = null;
  4. while (root != null) {
  5. if (root.left != null) {// 当左节点不为空时
  6. predecessor = root.left;
  7. // predecessor节点就是当前 root节点向左走一步,然后一直向右走至无法走为止
  8. while (predecessor.right != null && predecessor.right != root) {
  9. predecessor = predecessor.right;
  10. }
  11. // 让 predecessor 的右指针指向 root,继续遍历左子树
  12. if (predecessor.right != null) {
  13. predecessor.right = root;
  14. root = root.left;
  15. }
  16. // 说明左子树已经访问完了,我们需要断开链接
  17. else {
  18. res.add(root.val);
  19. predecessor.right = null;
  20. root = root.right;
  21. }
  22. }
  23. // 如果没有左孩子,则直接访问右孩子
  24. else {
  25. res.add(root.val);
  26. root = root.right;
  27. }
  28. }
  29. return res;
  30. }

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/50400
推荐阅读
相关标签
  

闽ICP备14008679号