二叉树—二叉树的前序遍历(leetcode 144)

最新推荐文章于 2025-05-07 11:10:49 发布

原创最新推荐文章于 2025-05-07 11:10:49 发布 · 73 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

89 篇文章

订阅专栏

这篇博客介绍了如何使用迭代算法解决二叉树的前序遍历问题。给定一个二叉树的根节点，算法通过操作空闲指针避免了递归，并在O(n)的时间复杂度和O(1)的空间复杂度下完成遍历。具体实现中，利用两个指针p和q，不断更新它们来遍历每个节点。当遇到没有左子树的节点时，直接访问其值并移动到右子节点；对于有左子树的节点，先将其右指针指向父节点，然后访问其值，最后调整节点的右指针来继续遍历。

题目描述

给你二叉树的根节点 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.com/problems/binary-tree-preorder-traversal/solution/dong-hua-mo-ni-er-cha-shu-qian-xu-die-da-q9yn/

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        TreeNode* p = root;
        TreeNode* q = nullptr;
        vector<int> ans;
        while(p) {
            q = p->left;
            if(q) {
                while(q->right != nullptr && q->right != p) {
                    q = q->right;
                }
                if(q->right == nullptr) {
                    ans.push_back(p->val);
                    q->right = p;
                    p = p->left;
                    continue;
                } else {
                    q->right = nullptr;
                }
            } else {
                ans.emplace_back(p->val);
            }
            p = p->right;
        }
        return ans;
    }
};