解法一:递归
「二叉树的先序遍历」的思路是:先访问根结点,再访问左子树,最后访问右子树;
「二叉树的中序遍历」的思路是:先访问左子树,再访问根结点,最后访问右子树;
「二叉树的后序遍历」的思路是:先访问左子树,再访问右子树,最后访问根结点;
下图以「先序遍历」为例进行展示:
基于上述思路,可以得到实现代码如下:
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
class Solution {
public:
/**
*
* @param root TreeNode类 the root of binary tree
* @return int整型vector<vector<>>
*/
vector<vector<int> > res;
vector<int> preorder, inorder, postorder;
vector<vector<int> > threeOrders(TreeNode* root) {
preOrder(root);
inOrder(root);
postOrder(root);
res.push_back(preorder);
res.push_back(inorder);
res.push_back(postorder);
return res;
}
void preOrder(TreeNode* root) {
if (!root) return;
preorder.push_back(root->val); // 访问根结点
preOrder(root->left); // 访问左子树
preOrder(root->right); // 访问右子树
}
void inOrder(TreeNode* root) {
if (!root) return;
inOrder(root->left); // 访问左子树
inorder.push_back(root->val); // 访问根结点
inOrder(root->right); // 访问右子树
}
void postOrder(TreeNode* root) {
if (!root) return;
postOrder(root->left); // 访问左子树
postOrder(root->right); // 访问右子树
postorder.push_back(root->val); // 访问根结点
}
}; 该方法需要遍历二叉树的所有结点,因此时间复杂度为O(N);
该方法在递归过程中需要使用栈空间,在最坏情况下(二叉树退化为链表),空间复杂度为O(N)。
解法二:非递归
二叉树的三种遍历也可以通过「非递归」的方式实现,其中,先序遍历与中序遍历的非递归实现方式类似,后序遍历实现方式有所不同。
先序遍历过程如图所示,步骤如下:
- 根结点入栈;
重复如下操作:- 访问栈顶结点;
- 右孩子入栈(若有);
- 左孩子入栈(若有);
中序遍历过程如图所示,步骤如下:
- 访问以当前结点p为根结点的「最左孩子」n;
- 访问n,并将p更新至其右子树上:
p = p->right; - 若p无右子树,则说明p是某棵子树的根结点,则在下一次循环中访问p;若p有右子树,则按照上述相同步骤访问该右子树,最后访问p。
后序遍历过程如图所示,步骤如下:
- 定义last指针,用以代表「上次访问的位置」,p结点用来遍历;
- 由于后续遍历需要先访问左子树和右子树、最后访问根结点,因此在满足如下条件时才能访问该结点:
- 该结点没有右子树;
- 该结点有右子树,且上次访问的结点(last指针)为其右子树,即该右子树已经被访问过了;
- 在不满足上述条件的情况下,说明当前结点还不能被访问,故先访问其右子树:
p = p->right; - 将p结点置为NULL,防止下次while循环时重复访问;
基于上述思路,实现代码如下:
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
class Solution {
public:
/**
*
* @param root TreeNode类 the root of binary tree
* @return int整型vector<vector<>>
*/
vector<vector<int> > res;
vector<int> preorder, inorder, postorder;
vector<vector<int> > threeOrders(TreeNode* root) {
preOrder(root);
inOrder(root);
postOrder(root);
res.push_back(preorder);
res.push_back(inorder);
res.push_back(postorder);
return res;
}
void preOrder(TreeNode* root) {
if (!root) return;
stack<TreeNode*> s;
TreeNode* p = root;
s.push(p);
while (s.size()) {
p = s.top(); // 访问根结点
s.pop(); // 根结点出栈
preorder.push_back(p->val);
if (p->right) s.push(p->right); // 右子树入栈
if (p->left) s.push(p->left); // 左子树入栈
}
return;
}
void inOrder(TreeNode* root) {
if (!root) return;
stack<TreeNode*> s;
TreeNode* p = root;
while (s.size() || p) {
while (p) { // 寻找最左的孩子
s.push(p);
p = p->left;
}
p = s.top();
inorder.push_back(p->val); // 访问
s.pop();
p = p->right; // 右子树
}
return;
}
void postOrder(TreeNode* root) {
if (!root) return;
stack<TreeNode*> s;
TreeNode* p = root, *last = NULL; // last记录先前被访问的结点
while (s.size() || p) {
while (p) { // 最左的孩子
s.push(p);
p = p->left;
}
p = s.top();
if (!p->right || last == p->right) { // 若该结点没有右孩子,或上一次访问的是右子树,则直接访问该结点
s.pop();
postorder.push_back(p->val);
last = p; // 更新last
p = NULL; // p置空,防止下一次循环重复访问
} else {
p = p->right; // 右子树
}
}
}
}; 该方法需要遍历二叉树的所有结点,因此时间复杂度为O(N);
该方法需要定义、使用栈空间,空间复杂度为O(N)。



京公网安备 11010502036488号