给定一个二叉树,返回它的 前序 遍历。
示例:
输入: [1,null,2,3] 1 \ 2 / 3 输出: [1,2,3]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
思路:
这道题有递归和非递归,先讲递归算法。
递归解法:在函数的参数中引入vector的引用,按照前序遍历(根-左-右),访问到当前节点就累加到vector中。最后返回vector
递归代码:
void preorderTraversalCore(TreeNode* root, vector<int> &res) {
if (!root) {
return;
}
res.push_back(root->val);
preorderTraversalCore(root->left,res);
preorderTraversalCore(root->right,res);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if (!root) {
return res;
}
preorderTraversalCore(root, res);
return res;
}
方法二(迭代代码):
这也是题目要求的前序遍历的非递归方法,我们需要掌握。
迭代遍历前序序列需要引入一个栈(先进后出特性),每次访问当前节点时压入栈,如果当前节点有左节点就不断压入栈,如果没有左节点意味着已经到最左叶节点,这时取栈顶元素pop出,如果pop的元素有右节点就压入stack,然后不断循环即可。
代码如下:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if (!root) {
return res;
}
stack<TreeNode*> stack;
TreeNode* tmp = root;
while (tmp != nullptr || !stack.empty()) {
while (tmp) {
res.push_back(tmp->val);
stack.push(tmp);
tmp = tmp->left;
}
if (!stack.empty()) {
tmp = stack.top();
stack.pop();
tmp = tmp->right;
}
}
return res;
}