力扣 二叉树的中序遍历

用了递归遍历,关于树的经典例题。

题目

递归

常规做法即递归了,不会写也得背下来。递归可以大致理解方法调用自身,先写中序遍历递归的方法,递归一定要有递归出口,当遍历到节点为空时返回,即已经找到了。可以看一下为什么是这三行,中序遍历为左中右顺序,那就可以先从根节点一直往左边找,直到找到最左边的节点,这是"递",然后到了这个节点时第一个inorder就会return,即开始"归"了,返回上一个inorder是不是就是该节点了,直接下一步add将当前节点值加进list集,然后下一个inorder就是遍历右边的节点了,当然这时右边有节点也会一直遍历下去,然后这里的顺序还是符合中序遍历,因为每次执行add时都是把当前节点为根节点,这样递归反复,在当前节点的左节点会在上一步返回先,在当前节点的右节点也会在该节点进入list集后在下一步返回。然后在另一个方法引入,返回list集即可。

时间复杂度:O(n),空间复杂度:O(n)。

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        inorder(root, res);
        return res;
    }

    public void inorder(TreeNode root, List<Integer> res) {
        if (root == null) {
            return;
        }
        inorder(root.left, res);
        res.add(root.val);
        inorder(root.right, res);
    }
}

循环

那要是不用递归怎么写,上述其实也可以看作是一个模拟栈,递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来,用循环能让这个栈更明显。先用外循环看看这个栈是否为空,节点非空也即要遍历的节点,然后下一个节点又是节点非空,把当前节点压入栈,指针左移,不断找左节点入栈,直到节点空了即找不到了,该循环结束,然后就开始出栈。出栈时位于栈顶的元素先出,即最左的元素先出,加进结果集后,指针右移,继续寻找右边的节点看看能否进行出栈,然后如此反复,就又是一个中序遍历了,思路是跟递归是差不多的。不想写多一个whlie,就把while改为if然后后面几行用else括住也能达到类似的效果。

时间复杂度:O(n),空间复杂度:O(n)。

java 复制代码
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        Deque<TreeNode> stk = new LinkedList<TreeNode>();
        while (root != null || !stk.isEmpty()) {
            while (root != null) {
                stk.push(root);
                root = root.left;
            }//一直往左找

            root = stk.pop();
            res.add(root.val);
            root = root.right;//指针右移
        }
        return res;
    }
}

Morris 中序遍历

不使用任何辅助空间,用上前驱节点,省去了栈的空间复杂度。先把根节点及根节点的右节点部分看成一大块连到根节点的左节点部分的最右节点上,然后以这样的方式反复拆分,左节点就会在前面先遍历,像链表一样的顺序,不过改变了整个树的结构。

时间复杂度:O(n),空间复杂度:O(1)。

java 复制代码
class Solution {
	public List<Integer> inorderTraversal(TreeNode root) {
		List<Integer> res = new ArrayList<Integer>();
		TreeNode pre = null;
		while(root!=null) {
			//如果左节点不为空,就将当前节点连带右子树全部挂到
			//左节点的最右子树下面
			if(root.left!=null) {
				pre = root.left;
				while(pre.right!=null) {
					pre = pre.right;
				}
				pre.right = root;
				//将root指向root的left
				TreeNode tmp = root;
				root = root.left;
				tmp.left = null;
			//左子树为空,则打印这个节点,并向右边遍历	
			} else {
				res.add(root.val);
				root = root.right;
			}
		}
		return res;
	}
}

当不想改变树的结构时,也可以进行链表的模拟,当遍历完后,将前驱节点的指针恢复即可。

java 复制代码
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        TreeNode pre = null;

        while (root != null) {
            if (root.left != null) {
                // pre 节点就是当前 root 节点向左走一步,然后一直向右走至无法走为止
                pre = root.left;
                while (pre.right != null && pre.right != root) {
                    pre = pre.right;
                }
                
                // 让 pre 的右指针指向 root,继续遍历左子树
                if (pre.right == null) {
                    pre.right = root;
                    root = root.left;
                }
                // 说明左子树已经访问完了,我们需要断开链接
                else {
                    res.add(root.val);
                    pre.right = null;
                    root = root.right;
                }
            }
            // 如果没有左孩子,则直接访问右孩子
            else {
                res.add(root.val);
                root = root.right;
            }
        }
        return res;
    }
}

像树这种结构很适合用递归循环实现。

相关推荐
数据小小爬虫几秒前
Jsoup解析商品详情时,如何确保数据准确性?
java·爬虫
互联网杂货铺2 分钟前
软件测试之压力测试
自动化测试·软件测试·测试工具·jmeter·职场和发展·测试用例·压力测试
V+zmm1013411 分钟前
自驾游拼团小程序的设计与实现(ssm论文源码调试讲解)
java·数据库·微信小程序·小程序·毕业设计
wen__xvn22 分钟前
每日一题洛谷P1914 小书童——凯撒密码c++
数据结构·c++·算法
坚定信念,勇往无前23 分钟前
springboot单机支持1w并发,需要做哪些优化
java·spring boot·后端
丁总学Java34 分钟前
`AdminAdminDTO` 和 `userSession` 对象中的字段对应起来的表格
java
BUG 劝退师1 小时前
八大经典排序算法
数据结构·算法·排序算法
m0_748240252 小时前
SpringMVC详解
java
m0_748240912 小时前
SpringMVC 请求参数接收
前端·javascript·算法
剑走偏锋o.O2 小时前
Java四大框架深度剖析:MyBatis、Spring、SpringMVC与SpringBoot
java·spring boot·spring·mybatis