原题链接:https://oj.leetcode.com/problems/binary-tree-preorder-traversal/
题目大意:先序遍历二叉树
思路:该算法属于基础算法,需要掌握其递归和非递归实现。面试时,要做到一次bug-free。递归思路和实现很简单,这里就不说了,下面主要介绍先序遍历的非递归实现算法。
我们用一个栈来模拟递归过程,若结点不空,则访问并入栈;若节点为空,但栈不空,则弹出栈顶元素,并访问其右子树。栈用于回溯。
时间复杂度:因为每个节点访问一次,所以时间复杂度为O(N)。 空间复杂度为树的深度 O(lgN)。
class Solution {
public:
vector<int> preorderTraversal(TreeNode *root) {
vector<int> res;
if(!root)
return res;
stack<TreeNode *> s;
while(root||!s.empty()){
if(root){
s.push(root);
res.push_back(root->val);
root=root->left;
}
else{
root=s.top();
s.pop();
root=root->right;
}
}
return res;
}
};
熟练掌握,做到bug free。树的遍历是许多与树相关的算法基础,所以应该重视。