二叉树的前序遍历

前言

在讲二叉树的遍历之前,我们首先讲一下递归这个东西,很多人,写递归函数靠运气,或者说不只是写递归函数,写任何代码都靠运气,听天由命,有时候自己都不知道自己写的是什么就过了,这样写代码肯定是不行的。

之所以我们会有这种情况是因为,我们在写一个代码的时候没有一个明确的方法体,我们在这里以递归为例子。

递归

我们在写递归算法的时候主要有三要素,每次写递归按照这三要素写,一般不会出错。

1.确定递归函数的参数和返回值: 确认哪些参数是递归过程中需要处理的,那么就在递归函数里加上这个参数,返回值每次递归的返回值一样,返回值很好确定,根据题意来决定。

2.确定终止条件: 我们在写递归函数时经常遇到的错误就是栈溢出,这说明你没有终止条件或者终止条件错误,操作系统也是用栈结构来保存第一次递归的信息。

3.确认单层递归的逻辑: 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归过程。

前序遍历

我们刚刚说道递归三要素,接下来我们就拿前序遍历实践一下。

在这之前我们先说明一下,什么是前序遍历

二叉树的前序遍历

二叉树前序遍历的顺序是根左右,那么上图这棵树的遍历顺序就是ABDECFG

144. 二叉树的前序遍历(点击立即答题

给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

示例 1:
二叉树的前序遍历

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

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

输入:root = [1]
输出:[1]
示例 4:

二叉树的前序遍历

输入:root = [1,2]
输出:[1,2]
示例 5:
二叉树的前序遍历

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

思路就是递归

java代码:

class Solution {
    
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<Integer>();
        //进入递归函数
        preorder(root , list);
        return list;
    }
	//构建递归函数,确认参数和返回值
    public void preorder(TreeNode root , List<Integer> result) {
    	//确定终止条件
        if (root == null) return;
		
		//将结果加入线性表list中
        result.add(root.val);
        //左节点
        preorder(root.left , result);
        //右节点
        preorder(root.right , result);
    }
}

至于单层逻辑就是整个递归函数了,你有了前两个条件才会有第三个条件。

或许还有同学不明白递归,我在这里解释一下,注意看注释;

	public void preorder(TreeNode root , List<Integer> result) {// 这里传过来两个参数的地址
	//如果传入的节点是null,则直接返回上一层
	//比如我们现在进入的的是第三层,那么我们就是返回第二层,并且第二层你是从哪进入第三层,你就返回到哪
    if (root == null) return;

    result.add(root.val);
 	 //再次传入两个参数的地址
 	 //假设我们从这里进入又一个递归函数
    preorder(root.left , result);
 
    preorder(root.right , result);
}

preorder(root.left , result);我们进入这个递归函数

	public void preorder(TreeNode root , List<Integer> result) {// 这里传过来的是root.left和result的地址
	
	//如果root.left == null ,那么直接就返回到preorder(root.left , result);的下一个代码,就是 preorder(root.right , result);之后就会执行preorder(root.right , result);

    if (root == null) return;

    result.add(root.val);
    preorder(root.left , result);
    preorder(root.right , result);
}

对于二叉树的遍历,我们还可以使用迭代法来遍历;

思路

迭代法其实就是将栈模拟出来,而递归是隐藏起来的。

在这里唯一需要注意的一点就是,前序遍历的顺序是根左右,我们在往栈中放的时候先放右节点,再放左节点,这是因为栈是先进后出的。
二叉树的前序遍历注:图片来源微信公众号代码随想录

java代码:(请看注释

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<Integer>(); 
        //若根节点为空,直接返回
        if (root == null) return result;
		
		//构建一个栈
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(root);
        //当栈不为空的情况下执行
        while (!stack.isEmpty()) {
        	//将栈顶元素赋予弄得
           TreeNode node = stack.peek();
           //弹出栈顶元素
           stack.pop();
           //因为我们不确定当前节点的子节点是否为空,所以在这里要进行判断
           if (node != null) result.add(node.val);
           else continue;
			
			//将当前节点的右节点放入栈
           stack.push(node.right);
           //将当前节点的左节点放入栈
           stack.push(node.left);
        }
       

        return result;
   
    }
}

若有误,请指教!

上一篇:一个可扩展的深度学习框架的Python实现(仿keras接口)


下一篇:由前序遍历和中序遍历构造二叉树