Day11 | 二叉树递归遍历迭代遍历统一迭代层序遍历

086小包字

已于 2024-07-15 23:35:54 修改

阅读量544

点赞数 24

文章标签：算法数据结构 java leetcode 面试

于 2024-07-15 17:38:47 首次发布

本文链接：https://blog.csdn.net/q864508127/article/details/140435463

版权

语言

Java

递归遍历

递归说明

递归有三要素：

1.确定递归函数的参数和返回值

2.确定终止条件

3.确定单层递归的逻辑

按照这三点来讲基本不会错。

来三道题目练练手。

题目

二叉树前序遍历

二叉树后续遍历

二叉树中序遍历

思路

讲一道中序遍历，创建一个ArrayList，使用递归函数，递归函数的参数是树的节点和结果。

在方法中先判断终止条件即节点不为空，因为是中序就左中右，先调用递归函数参数是左节点、ArrayList，再添加节点的值到ArrayList中，最后再调用递归函数参数是右节点、ArrayList。

最后返回结果ArrayList。

代码

// 前序遍历·递归·LC144_二叉树的前序遍历
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<Integer>();
        preorder(root, result);
        return result;
    }

    public void preorder(TreeNode root, List<Integer> result) {
        if (root == null) {
            return;
        }
        result.add(root.val);
        preorder(root.left, result);
        preorder(root.right, result);
    }
}
// 中序遍历·递归·LC94_二叉树的中序遍历
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        inorder(root, res);
        return res;
    }

    void inorder(TreeNode root, List<Integer> list) {
        if (root == null) {
            return;
        }
        inorder(root.left, list);
        list.add(root.val);             // 注意这一句
        inorder(root.right, list);
    }
}
// 后序遍历·递归·LC145_二叉树的后序遍历
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        postorder(root, res);
        return res;
    }

    void postorder(TreeNode root, List<Integer> list) {
        if (root == null) {
            return;
        }
        postorder(root.left, list);
        postorder(root.right, list);
        list.add(root.val);             // 注意这一句
    }
}

易错点

前序和后序遍历是一样的思路，更改递归函数的顺序就好了。

易错的就是考虑终止条件和递归参数。

迭代遍历

题目

二叉树中序遍历

二叉树前序遍历

二叉树后序遍历

主要讲讲前序遍历和中序遍历

前序遍历思路

采用栈的方式来模拟，将结果存在数组里。

首先创建一个数组，判断根节点是否为空，为空返回空数组。

再建立一个栈，将根节点添加到数组中，开始遍历终止条件是栈为空。

在循环里：先将根节点弹出，值放入数组中，判断如果右节点不为空放入右节点，再判断左节点，不为空把左节点放到栈中。这时小伙伴可能疑惑了，难道前序不是中左右吗这里为啥先加右节点？

因为栈是先入后出的。最后返回数组。

后序遍历思路

就是把中间的步骤改为先放左节点后放右节点，再把数组翻转。

前序遍历和后序遍历代码

class Solution {//前序遍历
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            res.add(node.val);
            if (node.right != null) {
                stack.push(node.right);
            }
            if (node.left != null) {
                stack.push(node.left);
            }
        }
        return res;
    }
}

class Solution {//后序遍历
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            res.add(node.val);
            if (node.left != null) {
                stack.push(node.left);
            }
            if (node.right != null) {
                stack.push(node.right);
            }
        }
        Collections.reverse(res);
        return res;
    }
}

中序遍历思路

中序遍历顺序左中右入栈顺序左右

首先创建一个数组，判断根节点是否为空，为空返回空数组。

再建立一个栈，将根节点添加到数组中，开始遍历终止条件是栈为空或节点为空。

判断节点是否为空，若不为空，将节点加入到栈中，节点向左子树遍历。

若为空，先弹出栈中的元素，再把元素的值加入到数组中，节点向右子树遍历。

最后返回结果数组。

中序遍历代码

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;
        while (cur != null || !stack.isEmpty()) {
            if (cur != null) {
                stack.push(cur);
                cur = cur.left;
            } else {
                cur = stack.pop();
                res.add(cur.val);
                cur = cur.right;
            }
        }
        return res; 
    }
}

统一迭代

思路

针对三种遍历，使用迭代法可以写出统一风格的代码！

依然采取栈的数据结构，我们将访问的节点放入栈中，把要处理的节点也放入栈中但是要做标记。

就是处理的节点放入栈后，紧接着放入一个空指针作为标记。

代码

中序遍历

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();//创建数组
        Stack<TreeNode> stack = new Stack<>();//创建栈
        if (root != null) stack.push(root);//判断根节点是否为空
        while (!stack.empty()) {//循环遍历二叉树
            TreeNode node = stack.peek();
            if (node != null) {
                stack.pop();//将节点弹出，避免重复操作
                if (node.right != null) stack.push(node.right);//添加右节点
                stack.push(node);// 添加中节点
                stack.push(null);//访问过但没处理，做一个标记
                if (node.left != null) stack.push(node.left);
            } else {
                stack.pop();
                node = stack.peek();
                stack.pop();
                res.add(node.val);
            }
        }
        return res;
    }
}

前序遍历

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();//创建数组
        Stack<TreeNode> stack = new Stack<>();//创建栈
        if (root != null) stack.push(root);//判断根节点是否为空
        while (!stack.empty()) {//循环遍历二叉树
            TreeNode node = stack.peek();
            if (node != null) {
                stack.pop();//将节点弹出，避免重复操作
                if (node.right != null) stack.push(node.right);//添加右节点
                if (node.left != null) stack.push(node.left);//左节点
                stack.push(node);// 添加中节点
                stack.push(null);//访问过但没处理，做一个标记
            } else {
                stack.pop();
                node = stack.peek();
                stack.pop();
                res.add(node.val);
            }
        }
        return res;
    }
}

后序遍历

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
              List<Integer> res = new ArrayList<>();//创建数组
        Stack<TreeNode> stack = new Stack<>();//创建栈
        if (root != null) stack.push(root);//判断根节点是否为空
        while (!stack.empty()) {//循环遍历二叉树
            TreeNode node = stack.peek();
            if (node != null) {
                stack.pop();//将节点弹出，避免重复操作
                stack.push(node);// 添加中节点
                stack.push(null);//访问过但没处理，做一个标记
                if (node.right != null) stack.push(node.right);//添加右节点
                if (node.left != null) stack.push(node.left);//左节点
            } else {
                stack.pop();
                node = stack.peek();
                stack.pop();
                res.add(node.val);
            }
        }
        return res;
    }
}

层序遍历

思想

想到了两种方式，递归和非递归。

一层一层的遍历，将每一层的个数存起来，放在数组中。

最后都放在二维数组中。

代码

递归方式

class Solution {
    public List<List<Integer>> resList = new ArrayList<List<Integer>>();//用来保存层序遍历的结果

    public List<List<Integer>> levelOrder(TreeNode root) {
        check01(root, 0);//调用方法 参数为节点和0
        return resList;//返回二维数组
    }
    public void check01(TreeNode node, Integer deep) {//传入节点 和 深度
        if (node == null) return;//判断节点是否为空
        deep++;//深度加一

        if (resList.size() < deep) {//如果数组的长度小于二叉树深度
            List<Integer> item = new ArrayList<Integer>();//创建一个一维数组
            resList.add(item);//将一维数组放在二维数组中
        }
        resList.get(deep - 1).add(node.val);//添加深度- 1节点的值
        check01(node.left, deep);//向左子树递归
        check01(node.right, deep);//向右子树递归
    }
}

迭代方式-借助队列

class Solution {
    public List<List<Integer>> resList = new ArrayList<List<Integer>>();//用来保存层序遍历的结果

    public List<List<Integer>> levelOrder(TreeNode root) {
        check02(root);//调用方法 参数为节点和0
        return resList;//返回二维数组
    }
    public void check02(TreeNode node) {//传入节点 
        if (node == null) return;//判断节点是否为空
        Queue<TreeNode> que = new LinkedList<TreeNode>();//创建队列
        que.offer(node);//把节点添加到队列中

        while (!que.isEmpty()) {//判断条件为队列不为空
            List<Integer> itemList = new ArrayList<Integer>();//创建一维数组
            int len = que.size();//获取队列的长度
            while (len > 0) {
                TreeNode tmpNode = que.poll();//弹出队列中的节点并赋值给临时节点
                itemList.add(tmpNode.val);//在一维数组中添加临时节点的值
                if (tmpNode.left != null) que.offer(tmpNode.left);//判断 加入左子节点
                if (tmpNode.right != null) que.offer(tmpNode.right);//判断 加入右字节点
                len--;//循环终止条件 让他遍历一层的数量大小
            }
            resList.add(itemList);//将一维数组添加到二维数组中。
        }
        
    }
}

总结

二叉树这里我感觉非常有意思，而且完成它花费了很长时间。继续努力距离大厂梦又进一步。

086小包字

关注

24
点赞
踩
10

收藏

觉得还不错? 一键收藏
0
评论
Day11 | 二叉树递归遍历迭代遍历统一迭代层序遍历

在方法中先判断终止条件即节点不为空，因为是中序就左中右，先调用递归函数参数是左节点、ArrayList，再添加节点的值到ArrayList中，最后再调用递归函数参数是右节点、ArrayList。在循环里：先将根节点弹出，值放入数组中，判断如果右节点不为空放入右节点，再判断左节点，不为空把左节点放到栈中。依然采取栈的数据结构，我们将访问的节点放入栈中，把要处理的节点也放入栈中但是要做标记。判断节点是否为空，若不为空，将节点加入到栈中，节点向左子树遍历。首先创建一个数组，判断根节点是否为空，为空返回空数组。
复制链接

扫一扫