leetcode 144. Binary Tree Preorder Traversal

最新推荐文章于 2024-05-30 07:40:02 发布

原创最新推荐文章于 2024-05-30 07:40:02 发布 · 287 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

26 篇文章

订阅专栏

二叉树

4 篇文章

订阅专栏

本文介绍二叉树前序遍历的两种方法：使用栈实现和模拟递归过程。详细解释了每种方法的具体步骤及其实现代码。

Discription

Given a binary tree, return the preorder traversal of its nodes’ values.

For example:
Given binary tree {1,#,2,3},
1
\
2
/
3
return [1,2,3].

思路

方法一

先让根进栈，只要栈不为空，就可以做弹出操作。
每次弹出一个结点，把它的左右结点都进栈，其中右子树先进栈，这样可以保证右子树在栈中总处于左子树的下面。

时间复杂度： $?$
空间复杂度： $?$

代码

vector<int> preorderTraversal(TreeNode* root) {
    vector<int> res;
    vector<TreeNode*> sta;

    if (root)
        sta.push_back(root);
    while (!sta.empty())
    {
        TreeNode *pNode = sta.back();

        res.push_back(pNode->val);
        sta.pop_back();

        if (pNode->right) sta.push_back(pNode->right);
        if (pNode->left) sta.push_back(pNode->left);
    }

    return res;
}

方法二

每次都向左走到头，然后再向右走，如图：

时间复杂度： $?$
空间复杂度： $?$

代码

vector<int> preorderTraversal(TreeNode* root) {
    vector<int> res;
    vector<TreeNode*> sta;

    if (root) sta.push_back(root);
    while (!sta.empty())
    {
        while (sta.back())
        {
            //访问栈顶
            res.push_back(sta.back()->val);

            //向左走
            sta.push_back(sta.back()->left);
        }

        // 弹出空指针
        sta.pop_back();

        //向右走
        if (!sta.empty())
        {
            TreeNode *pNode = sta.back();
            sta.pop_back();

            sta.push_back(pNode->right);
        }
    }

    return res;
}