LeetCode——二叉树的非递归遍历

简介: LeetCode——二叉树的非递归遍历

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]

提示

树中节点数目在范围 [0, 100] 内

-100 <= Node.val <= 100

进阶

递归算法很简单,你可以通过迭代算法完成吗?

原题链接:https://leetcode.cn/problems/binary-tree-preorder-traversal/

思路

将一棵树分成左路节点和左路节点的右子树。

那么迭代如何实现呢?这需要一个栈来储存结点。

先从头结点,访问到左子树的空为止,但是再这个过程中要将经过的数放进一个数组中。

不停的去访问左子树,到空就取栈顶然后pop掉,让结点指针指向栈顶结点的右子树就可以了,以此循环。

代码

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int>arr;//储存前序遍历的结果
        stack<TreeNode*>st;
        TreeNode* cur = root;
        while(!st.empty() || cur)//如果栈为空,并且cur指向了空,因为栈为空,cur指向的右子树说不定还有值
        {
            while(cur)
            {
                arr.push_back(cur->val);//将结点值添加到arr中
                st.push(cur);//入栈
                cur = cur->left;
            }
            TreeNode* top = st.top();//取栈顶结点
            st.pop();//出栈
            cur = top->right;//访问栈顶元素的右子树
        }
        return arr;
    }
};

94. 二叉树的中序遍历

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

示例 1

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

输出:[1,3,2]

示例 2

输入:root = []

输出:[]

示例 3

输入:root = [1]

输出:[1]

提示

树中节点数目在范围 [0, 100] 内

-100 <= Node.val <= 100

进阶:

递归算法很简单,你可以通过迭代算法完成吗?

原题链接:https://leetcode.cn/problems/binary-tree-inorder-traversal/

思路

中序遍历和前序遍历没什么太大差别,只是插入数据的位置不同而已,要在出栈的时候访问栈顶的元素。

代码

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int>arr;//储存中序遍历的结果
        stack<TreeNode*>st;
        TreeNode* cur = root;
        while(!st.empty() || cur)
        {
            while(cur)
            {
                st.push(cur);//入栈
                cur = cur->left;
            }
            TreeNode* top = st.top();//取栈顶结点
            arr.push_back(top->val);//将结点值添加到arr中
            st.pop();//出栈
            cur = top->right;//访问栈顶元素的右子树
        }
        return arr;
    }
};

145. 二叉树的后序遍历

给你一棵二叉树的根节点 root ,返回其节点值的后序遍历。

示例 1

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

输出:[3,2,1]

示例 2

输入:root = []

输出:[]

示例 3

输入:root = [1]

输出:[1]

提示

树中节点的数目在范围 [0, 100] 内

-100 <= Node.val <= 100

进阶

递归算法很简单,你可以通过迭代算法完成吗?

原题链接:https://leetcode.cn/problems/binary-tree-postorder-traversal/

思路

利用栈的基本思路是不变的,在弹出的时候,加一个判断条件,判断该节点的左子树和右子树是否全都访问完毕,如果访问完毕才能放进数组中。

这里只需要加一个结点指针指向栈最后一个弹出的结点就可以了:

这里15结点已经从栈中弹出去了,然后是7也弹出去。

front就会指向cur的位置:

这个时候cur检测右子树是不是front就能知道右子树是不是已经被访问完毕了。

代码

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int>arr;//储存后序遍历的结果
        stack<TreeNode*>st;
        TreeNode* cur = root;
        TreeNode* front = nullptr;//记录最后上一个出栈的结点位置
        while(cur || !st.empty())
        {
            while(cur)
            {
                st.push(cur);
                cur = cur -> left;
            }
            TreeNode* top = st.top();
            //如果栈顶节点右子树为空就说明该节点的左右子树都访问完毕了。
            //如果栈顶节点的右子树等于上个弹出的结点,那么也说明这该节点左右子树都访问完毕了。
            if(top->right == nullptr || top->right == front)
            {
                st.pop();
                arr.push_back(top->val);
                front = top;
            }
            else
            {
                cur = top -> right;
            }
        }
        return arr;
    }
};

相关文章
|
1月前
|
存储 算法
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
文章深入探讨了二叉树的层序遍历方法,并展示了如何通过队列实现层序遍历的算法逻辑,同时指出掌握层序遍历技巧可以帮助解决LeetCode上的多道相关题目。
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
|
1月前
|
算法 Java
LeetCode第94题二叉树的中序遍历
文章介绍了LeetCode第94题"二叉树的中序遍历"的解法,使用递归实现了中序遍历的过程,遵循了"左根右"的遍历顺序,并提供了清晰的Java代码实现。
LeetCode第94题二叉树的中序遍历
|
1月前
|
存储
LeetCode------递归(爬楼梯)
这篇文章通过LeetCode上的"爬楼梯"问题介绍了递归的基本概念和实现方法,包括递归公式的推导、基本递归实现、使用备忘录优化以避免重复计算,以及自底向上的迭代方法来提高效率。
LeetCode------递归(爬楼梯)
|
1月前
|
存储 算法 Java
LeetCode经典算法题:二叉树遍历(递归遍历+迭代遍历+层序遍历)以及线索二叉树java详解
LeetCode经典算法题:二叉树遍历(递归遍历+迭代遍历+层序遍历)以及线索二叉树java详解
60 0
|
1月前
|
算法 Java
LeetCode初级算法题:子数组最大平均数+二叉树的最小深度+最长连续递增序列+柠檬水找零
LeetCode初级算法题:子数组最大平均数+二叉树的最小深度+最长连续递增序列+柠檬水找零
32 0
|
1月前
|
Python
【Leetcode刷题Python】剑指 Offer 32 - III. 从上到下打印二叉树 III
本文介绍了两种Python实现方法,用于按照之字形顺序打印二叉树的层次遍历结果,实现了在奇数层正序、偶数层反序打印节点的功能。
41 6
|
1月前
|
Python
【Leetcode刷题Python】剑指 Offer 26. 树的子结构
这篇文章提供了解决LeetCode上"剑指Offer 26. 树的子结构"问题的Python代码实现和解析,判断一棵树B是否是另一棵树A的子结构。
35 4
|
1月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
73 2
|
1月前
|
索引 Python
【Leetcode刷题Python】从列表list中创建一颗二叉树
本文介绍了如何使用Python递归函数从列表中创建二叉树,其中每个节点的左右子节点索引分别是当前节点索引的2倍加1和2倍加2。
36 7
|
1月前
|
Python
【Leetcode刷题Python】剑指 Offer 30. 包含min函数的栈
本文提供了实现一个包含min函数的栈的Python代码,确保min、push和pop操作的时间复杂度为O(1)。
17 4