牛客热题：二叉树的前序遍历

news2025/1/18 17:14:09

📟作者主页：慢热的陕西人

🌴专栏链接：力扣刷题日记

📣欢迎各位大佬👍点赞🔥关注🚓收藏，🍉留言

在这里插入图片描述

文章目录

牛客热题：二叉树的前序遍历
- 题目链接
- 方法一：递归实现
- - 思路
  - 代码
  - 复杂度
- 方法二：非递归
- - 思路
  - 代码
  - 复杂度

牛客热题：二叉树的前序遍历

题目链接

二叉树的前序遍历_牛客题霸_牛客网 (nowcoder.com)

方法一：递归实现

思路

使用递归的方法
先将根节点的值放入到对应的答案数组里
然后分别递归左子树和右子树

代码

    vector<int> res;
    void _preorder(TreeNode* root)
    {
        if(root == nullptr) return ;
        res.push_back(root->val);
        if(root->left != nullptr) _preorder(root->left);
        if(root->right != nullptr) _preorder(root->right);
        return ;
    }
    vector<int> preorderTraversal(TreeNode* root) 
    {
        _preorder(root);
        return res;
    }

复杂度

时间复杂度：O(N) ,遍历了一遍二叉树

空间复杂度：O(N) , 使用了一个和二叉树节点数相同的数组

方法二：非递归

思路

用栈模拟函数递归的栈帧
首先我们将根节点push到栈内部
然后我们将root的值push到res数组内部
再将右子树(如果存在)插入到对应的栈
左子树(如果存在)插入到对应的栈
直到对应的栈为空

代码

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

        if(root == nullptr) return res;

        stack<TreeNode*> st;
        st.push(root);
        while(!st.empty())
        {
            TreeNode* cur = st.top();
            st.pop();
            res.push_back(cur->val);
            if(cur->right != nullptr) st.push(cur->right);
            if(cur->left != nullptr) st.push(cur->left);
        }

        return res;
    }