携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第26天,点击查看活动详情
二叉树的前序遍历
题目描述:
给你二叉树的根节点 root ,返回它节点值的 前序 **遍历。
示例1:
输入: root = [1,null,2,3]
输出: [1,2,3]
示例2:
输入: root = []
输出: []
示例3:
输入: root = [1]
输出: [1]
示例4:
输入: root = [1,2]
输出: [1,2]
解题思路
首先了解下前序遍历:
二叉树的前序遍历:按照访问根节点——左子树——右子树的方式遍历这棵树
思路一: 递归
我们只要首先将 root 节点的值加入result,然后递归调用 preorderTraversalNode(root.left) 来遍历 root 节点的左子树,最后递归调用 preorderTraversalNode(root.right) 来遍历 root 节点的右子树即可,递归终止的条件为碰到空节点
实现代码如下:
var preorderTraversal = function (root) {
var result = [];
var preorderTraversalNode = function (node) {
if (node) {
// 先根节点
result.push(node.val);
// 左子树
preorderTraversalNode(node.left);
// 右子树
preorderTraversalNode(node.right);
}
}
preorderTraversalNode(root);
return result;
};
时间复杂度: O(n),其中 n 是二叉树的节点数
空间复杂度: O(n)
思路二: 迭代
利用栈来记录遍历的过程:
-
首先 根入栈;
-
根节点出栈,将根节点值存入数组中;
-
然后遍历左子树、右子树;因为栈是先进后出,所以右子树先入栈;
-
继续出栈(左子树出栈)
-
依次循环出栈遍历入栈,直到栈为空
与递归的区别:
递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来
实现代码如下:
var preorderTraversal = function (root) {
var result = [], stack = [];
if (root) stack.push(root);
while (stack.length) {
var curr = stack.pop();
result.push(curr.val);
if (curr.right !== null) {
stack.push(curr.right)
}
if (curr.left !== null) {
stack.push(curr.left)
}
}
return result;
};
时间复杂度:O(n),其中 n 是二叉树的节点数
空间复杂度:O(n)