赞
踩
给你二叉树的根节点 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]
提示:
进阶:递归算法很简单,你可以通过迭代算法完成吗?
首先我们需要了解什么是二叉树的前序遍历:按照访问根节点——左子树——右子树的方式遍历这棵树,而在访问左子树或者右子树的时候,我们按照同样的方式遍历,直到遍历完整棵树。因此整个遍历过程天然具有递归的性质,我们可以直接用递归函数来模拟这一过程。
定义 preorder(root) 表示当前遍历到 root 节点的答案。按照定义,我们只要首先将 root 节点的值加入答案,然后递归调用 preorder(root.left) 来遍历 root 节点的左子树,最后递归调用 preorder(root.right) 来遍历 root 节点的右子树即可,递归终止的条件为碰到空节点。
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode() : val(0), left(nullptr), right(nullptr) {} * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} * }; */ class Solution { vector<int> ans; public: vector<int> preorderTraversal(TreeNode* root) { preorder(root); return ans; } void preorder(TreeNode* root){ if(root == nullptr) return; ans.push_back(root->val); preorder(root->left); preorder(root->right); } };
时间复杂度:
O
(
n
)
O(n)
O(n),其中
n
n
n 是二叉树的节点数。每一个节点恰好被遍历一次。
空间复杂度:
O
(
n
)
O(n)
O(n),为递归过程中栈的开销,平均情况下为
O
(
log
n
)
O(\log n)
O(logn),最坏情况下树呈现链状,为
O
(
n
)
O(n)
O(n)。
我们也可以用迭代的方式实现方法一的递归函数,两种方式是等价的,区别在于递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来——实现其 根结点 > 左子树 > 右子树 的遍历顺序,其余的实现与细节都相同,
我们使用栈来进行迭代,过程如下:
由于栈是“先进后出”的顺序,所以入栈时先将右子树入栈,这样使得前序遍历结果为 “根->左->右”的顺序。
Leetcode 官方题解
class Solution { public: vector<int> preorderTraversal(TreeNode* root) { vector<int> res; if (root == nullptr) return res; stack<TreeNode*> stk; TreeNode* node = root; while (!stk.empty() || node != nullptr) { while (node != nullptr) { res.emplace_back(node->val); stk.emplace(node); node = node->left; } node = stk.top(); stk.pop(); node = node->right; } return res; } };
我的
class Solution { public: vector<int> preorderTraversal(TreeNode* root) { vector<int> ans; if(root == nullptr) return ans; stack<TreeNode*> nodes; // 进栈的结点,当再次访问时,即证明其左子树的结点均已访问完 bool recall= false; // true 回溯,false 不回溯 TreeNode* p = nullptr; nodes.push(root); ans.push_back(root->val); while(!nodes.empty()){ // recall 用于回溯时不重复遍历栈中结点的左子树 if(nodes.top()->left != nullptr && !recall){ // 若栈顶结点的左子树非空 ans.push_back(nodes.top()->left->val); nodes.push(nodes.top()->left); }else if(nodes.top()->right != nullptr){ // 若栈顶结点的右子树为空 recall = false; p = nodes.top()->right; ans.push_back(p->val); nodes.pop(); nodes.push(p); }else{ nodes.pop(); recall = true; // 遇到叶子结点,开始回溯,访问此时栈中结点的右子树 } } return ans; } };
时间复杂度:
O
(
n
)
O(n)
O(n),其中
n
n
n 是二叉树的节点数。每一个节点恰好被遍历一次。
空间复杂度:
O
(
n
)
O(n)
O(n),为迭代过程中显式栈的开销,平均情况下为
O
(
log
n
)
O(\log n)
O(logn),最坏情况下树呈现链状,为
O
(
n
)
O(n)
O(n)。
有一种巧妙的方法可以在线性时间内,只占用常数空间来实现前序遍历。这种方法由 J. H. Morris 在 1979 年的论文「Traversing Binary Trees Simply and Cheaply」中首次提出,因此被称为 Morris 遍历。
Morris 遍历的核心思想是利用树的大量空闲指针,实现空间开销的极限缩减。Morris 的前序遍历规则总结如下:
root
;这样我们利用 Morris 遍历的方法,前序遍历该二叉树,即可实现线性时间与常数空间的遍历。
class Solution { public: vector<int> preorderTraversal(TreeNode *root) { vector<int> res; if (root == nullptr) { return res; } TreeNode *p1 = root, *p2 = nullptr; while (p1 != nullptr) { p2 = p1->left; if (p2 != nullptr) { while (p2->right != nullptr && p2->right != p1) { p2 = p2->right; } if (p2->right == nullptr) { res.emplace_back(p1->val); p2->right = p1; p1 = p1->left; continue; } else { p2->right = nullptr; } } else { res.emplace_back(p1->val); } p1 = p1->right; } return res; } };
时间复杂度:
O
(
n
)
O(n)
O(n),其中
n
n
n 是二叉树的节点数。没有左子树的节点只被访问一次,有左子树的节点被访问两次。
空间复杂度:
O
(
1
)
O(1)
O(1)。只操作已经存在的指针(树的空闲指针),因此只需要常数的额外空间。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。