Binary Tree Preorder Traversal--LeetCode

1.题目

Binary Tree Preorder Traversal

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].
Note: Recursive solution is trivial, could you do it iteratively?

2.题意

二叉树前序遍历

3.分析

1）递归是最简单的方法
直接按照root、左、右的顺序放进vector即可
算法的时间复杂度是O(n)，而空间复杂度则是递归栈的大小，即O(logn)

2）借助栈来模拟递归
先将root结点进栈，然后按层序遍历
利用栈后进先出的特点，先让右孩子节点进栈，再让左孩子节点进栈
在每一个非空的下一层依次取出左右孩子节点
算法时间复杂度是O(n)，空间复杂度是栈的大小O(logn)

4.代码

1) 递归版

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;

        preorder(result, root);

        return result;
    }

private:
    void preorder(vector<int> &result, TreeNode *root) {
        if(root == nullptr)
            return;

        result.push_back(root->val);
        preorder(result, root->left);
        preorder(result, root->right);
    }
};

2) 迭代版

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        if(root == nullptr)
            return result;

        stack<const TreeNode*> s;
        if(root != nullptr)
            s.push(root);

        while(!s.empty())
        {
            const TreeNode *p = s.top();
            s.pop();
            result.push_back(p->val);

            if(p->right != nullptr)
                s.push(p->right);
            if(p->left != nullptr)
                s.push(p->left);
        }

        return result;
    }
};