144-二叉树的前序遍历

马谦马谦马谦

615
文章

17
评论

2020 年 2 月 8 日 17:32:42数据结构和算法评论180字数 253阅读 0 分 50 秒阅读模式

一、题目描述

给定一个二叉树，返回它的前序遍历结果。

例如输入二叉树 [1,null,2,3]：

输出：

[1,2,3]

[1,2,3]

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

二、题解

二叉树的前序遍历，递归和非递归方式。

参考：二叉树的前序遍历。

三、代码

3.1 递归代码

class Solution {
    void preorderTraversal(TreeNode *root, vector<int> &v) {
        if (root == nullptr) {
            return;
        }
        v.push_back(root->val);
        if (root->left) {
            preorderTraversal(root->left, v);
        }
        if (root->right) {
            preorderTraversal(root->right, v);
        }
    }
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> v;
        preorderTraversal(root, v);
        return v;
    }
};

class Solution {

void preorderTraversal(TreeNode *root, vector<int> &v) {

if (root == nullptr) {

return;

}

v.push_back(root->val);

if (root->left) {

preorderTraversal(root->left, v);

}

if (root->right) {

preorderTraversal(root->right, v);

}

public:

vector<int> preorderTraversal(TreeNode* root) {

vector<int> v;

preorderTraversal(root, v);

return v;

}

};

144-二叉树的前序遍历-图片1

3.2 非递归代码

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode *> st;
        TreeNode *p;
        vector<int> v;

        p = root;
        while (p || !st.empty()) {
            while (p) {
                // 遍历当前节点
                v.push_back(p->val);
                st.push(p);
                // 遍历左子树
                p = p->left;
            }

            // 通过栈来遍历右子树
            if (!st.empty()) {
                p = st.top()->right;
                st.pop();
            }
        }

        return v;
    }
};