leetcode144. 二叉树的前序遍历（迭代法）

最新推荐文章于 2023-11-13 20:27:20 发布

一条特立独行的狗、

最新推荐文章于 2023-11-13 20:27:20 发布

阅读量257

点赞数

分类专栏： Leetcode树文章标签：二叉树 leetcode 算法

本文链接：https://blog.csdn.net/weixin_43498985/article/details/108365892

版权

Leetcode树专栏收录该内容

29 篇文章 0 订阅

订阅专栏

给定一个二叉树，返回它的前序遍历。
示例:
输入: [1,null,2,3]
输出: [1,2,3]

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> vecRes;
    vector<int> preorderTraversal(TreeNode* root) {
        if(root==nullptr)
            return vecRes;
        stack<TreeNode *> s;
        TreeNode * pCurNode = root;
        while(!s.empty() || pCurNode)
        {
            if(pCurNode)
            {
                vecRes.push_back(pCurNode->val);
                s.push(pCurNode);
                pCurNode = pCurNode->left;
            }
            else
            {
                TreeNode * pNode = s.top();
                s.pop();
                pCurNode = pNode->right;
            }
        }
        return vecRes;
    }
};