给你二叉树的根节点 root ,返回它节点值的 前序 **遍历。
1
\
2
/
3
输入: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]
方法一:递归
思路与算法
首先我们需要了解什么是二叉树的前序遍历:按照访问根节点——左子树——右子树的方式遍历这棵树,而在访问左子树或者右子树的时候,我们按照同样的方式遍历,直到遍历完整棵树。因此整个遍历过程天然具有递归的性质,我们可以直接用递归函数来模拟这一过程。
定义 preorder(root) 表示当前遍历到 root 节点的答案。按照定义,我们只要首先将 root 节点的值加入答案,然后递归调用 preorder(root.left) 来遍历 root 节点的左子树,最后递归调用 preorder(root.right) 来遍历 root 节点的右子树即可,递归终止的条件为碰到空节点。
var preorderTraversal = function (root) {
let res = [];
preorder(root, res);
return res;
function preorder(root, res) {
if (!root) return;
res.push(root.val);
preorder(root.left, res)
preorder(root.right, res)
}
};
方法二:迭代
思路与算法
我们也可以用迭代的方式实现方法一的递归函数,两种方式是等价的,区别在于递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来,其余的实现与细节都相同,具体可以参考下面的代码。
var preorderTraversal = function (root) {
let res = [];
if (!root) return res;
let stack = [];
let node = root;
while (stack.length != 0 || node) {
while (node) {
res.push(node.val);
stack.push(node);
node = node.left
}
node = stack.pop();
node = node.right
}
return res;
};