审题
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,2,3]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
示例 4:
输入:root = [1,2]
输出:[1,2]
示例 5:
输入:root = [1,null,2]
输出:[1,2]
提示:
树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100
进阶:递归算法很简单,你可以通过迭代算法完成吗?
看到这道题,一看,递归实现直接就可以AC了,然而为了掌握递归的底层逻辑,我们还是要搞一下迭代试一试
- 在方案一中,递归。
- 在方案二中,迭代(递归的本质就是模拟一个栈)。
代码实现
方案一:
常规操作,没啥好说的。
class Solution {
public:
void pre(TreeNode* root, vector<int>& v) {
if (root) v.push_back(root->val);
else return;
pre(root->left, v);
pre(root->right, v);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> v;
pre(root, v);
return v;
}
};
方案2:
这个迭代,我有一个地方还是每太理解,就是标注释的地方,这个条件让我楞想可能想不太出来…
1:这个的整体思路就是第一个循环,直接遍历到最左下,这个最左下可以认为是二叉树的最左下的左节点(一个空结点)。
2:然后将二叉树最左下的结点pop出来
3:遍历右结点,如果右结点为非空,继续转到1。如果右结点为空结点,直接将该最左下的父结点pop掉,再转回3,以此类推。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> v;
if ( !root ) return v;
stack<TreeNode*> s;
while ( !s.empty() || root ) { //?
while ( root ) {
s.push(root);
v.push_back(root->val);
root = root->left;
}
root = s.top();
s.pop();
root = root->right;
}
return v;
}
};
反思
这个栈的知识还是比较浅薄,尤其是结合了树之后。因此,如果想要掌握好这部分的内容,我认为首要的不是求甚解,而是不求甚解,在极其细节的部分可以先记下来,以后随着练习的增多自然会加深理解,不必搞自己心态。