训练营十四天(二叉树的遍历)

二叉树的递归遍历

java 复制代码
树的定义
public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    
    public TreeNode() {}
    
    public TreeNode(int val) {
        this.val = val;
    }
    
    public TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

递归三要素

  • 确定递归函数的参数和返回值
  • 确定终止条件
  • 确定单层递归的逻辑

1.前序遍历 根左右

144.二叉树的前序遍历(opens new window)

java 复制代码
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<Integer>();
        preorder(root, result);
        return result;
    }

    private void preorder(TreeNode root,  List<Integer> result){
        if(root == null)
            return;
        result.add(root.val);
        preorder(root.left,result);
        preorder(root.right,result);
    }
}

2.后序遍历 左右根

145.二叉树的后序遍历(opens new window)

java 复制代码
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<Integer>();
        postorder(root, result);
        return result;
    }

    private void postorder(TreeNode root,  List<Integer> result){
        if(root == null)
            return;
        postorder(root.left,result);
        postorder(root.right,result);
        result.add(root.val);
    }
}

3. 中序遍历 左根右

94.二叉树的中序遍历

java 复制代码
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<Integer>();
        inorder(root, result);
        return result;
    }

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

二叉树的迭代遍历

1.前序遍历 根左右

使用栈,因为栈是后入先出,所以使用栈来记录结点,先押入跟结点,在押入右结点和左节点,按照出栈顺序访问元素

java 复制代码
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
		List<Integer> result = new ArrayList<>();
		if (root == null)
			return result;
		Stack<TreeNode> stack = new Stack<>();
		stack.push(root);
		while (!stack.isEmpty()){
			TreeNode temp = stack.pop();
			//left后入栈会先出栈,因为是根左右
			result.add(temp.val);
			if (temp.right != null)
				stack.push(temp.right);
			if (temp.left != null)
				stack.push(temp.left);
		}
		return result;
    }
}

2.后序遍历 左右根

与前序遍历基本一致

  • 后序遍历顺序 左-右-中
  • 入栈顺序:中-左-右
  • 出栈顺序:中-右-左
  • 最后翻转结果(一定要记得翻转结果,否则得到的是根右左)
java 复制代码
class Solution {
	public List<Integer> postorderTraversal(TreeNode root) {
		List<Integer> result = new ArrayList<>();
		if (root == null)
			return result;
		Stack<TreeNode> stack = new Stack<>();
		stack.push(root);
		while (!stack.isEmpty()){
			TreeNode temp = stack.pop();
			result.add(temp.val);
			if (temp.left != null){
				stack.push(temp.left);
			}
			if (temp.right != null){
				stack.push(temp.right);
			}
		}
		Collections.reverse(result);
		return result;
	}
}

3. 中序遍历 左根右

单独的思路,因为不能先将根压入栈,那么如果先将left入栈会导致指针的丢失,所以要先找到左子树为空的,也就是下一个元素就是当前的栈顶,即当前子树的根结点

java 复制代码
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
		List<Integer> result = new ArrayList<>();
		if (root == null)
			return result;
		TreeNode cur = root;
		Stack<TreeNode> stack = new Stack<>();
		while (!stack.isEmpty() || cur != null){
			if (cur != null){//此时要找到左子树的底部
				stack.push(cur);
				cur = cur.left;
			}else {//当前结点没有了左子树,那么当前没有左子树的跟结点就是栈顶
				cur = stack.pop();
				result.add(cur.val);
				cur = cur.right;
			}
		}
		return result;
    }
}
相关推荐
小莫分享1 小时前
github 镜像节点
java
链上Sniper1 小时前
智能合约状态快照技术:实现 EVM 状态的快速同步与回滚
java·大数据·linux·运维·web3·区块链·智能合约
缘来是庄2 小时前
设计模式之建造者模式
java·设计模式·建造者模式
小湘西2 小时前
Apache HttpClient 的请求模型和 I/O 类型
java·http·apache
沃夫上校2 小时前
Feign调Post接口异常:Incomplete output stream
java·后端·微服务
q567315232 小时前
Java Selenium反爬虫技术方案
java·爬虫·selenium
张小洛2 小时前
Spring IOC容器核心阶段解密:★Bean实例化全流程深度剖析★
java·后端·spring·ioc容器·bean实例化
不良手残2 小时前
IDEA类和方法注释模板设置-保姆教程
java·开发语言
GoodStudyAndDayDayUp2 小时前
调用海康API预览视频
java·海康
李迟3 小时前
在Linux服务器上使用kvm创建虚拟机
java·linux·服务器