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;
}
};