题目地址:力扣
二叉树的后序遍历就是, 左 -> 右 -> 根 ,这样的次序。
解法1:递归,这个模板是前中后通用的
class Solution {
public:
void postorder(TreeNode *root, vector<int> &res)
{
if (root != nullptr)
{
if (root->left != nullptr)
postorder(root->left, res);
if (root->right != nullptr)
postorder(root->right, res);
res.push_back(root->val);
}
}
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
postorder(root, res);
return res;
}
};
解法2:迭代(手动建栈模拟系统操作)
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
TreeNode *prev = nullptr;
// 使用prev存储上一个出栈的结点
stack<TreeNode*> stk;
while (root != nullptr || !stk.empty())
{
while (root != nullptr)
{
stk.push(root);
root = root->left;
}
// 保证从栈里面出来的都是只有右孩子没探索过的
root = stk.top();
// 下面的root都是从栈里出来的
// 若没有右孩子,则直接加入结果并出栈,并将root置空
if (root->right == nullptr )
{
prev = root;
res.push_back(root->val);
stk.pop();
root = nullptr;
}
// 若当前节点有右孩子,则需要判断当前节点的右孩子是否已经访问过了
else
{ // 如果当前节点的右孩子没访问过,那么就访问其右孩子
if (root->right != prev)
root = root->right;
// 如果当前节点的右孩子是上一个出栈的元素,则证明其右子树已经访问过
// 这里和前面的右孩子为空的时候的操作是一样的,两者可以用 || 连起来
// 但是从逻辑上来说的话应该是放这个位置更合适
else{
prev = root;
res.push_back(root->val);
stk.pop();
root = nullptr;
}
}
}
return res;
}
};
解法3:Morris遍历
二叉树后序遍历的Morris方法可以说是最难的,主要思想还是把左子树的最右节点指向根节点。然后回来的时候把指针给取消。但是不同的地方就在于打印的时候要么是只打印左孩子单个叶子节点,要么就是打印根节点及其右孩子叶子节点,因此每次打印的时候需要使用reverse来反转一下,而且最后树的总的根节点需要单独打印,所以最后会对root进行单独的打印操作
class Solution {
public:
// addPath
void addPath(vector<int> &vec, TreeNode *node) {
int count = 0;
while (node != nullptr) {
++count;
vec.push_back(node->val);
node = node->right;
}
reverse(vec.end() - count, vec.end());
}
vector<int> postorderTraversal(TreeNode *root) {
vector<int> res;
TreeNode *p1 = root, *p2 = nullptr;
// p2的设定是p1的左孩子
// p1非空就继续循环
while (p1 != nullptr) {
p2 = p1->left;
// 若p1有左孩子
if (p2 != nullptr) {
// p2是p1左子树的最右节点
while (p2->right != nullptr && p2->right != p1) {
p2 = p2->right;
}
// p2->right == nullptr 确定了该节点是第一次访问
// 先把p1加入结果,再把p1左子树的最右节点指向p1,p1继续往左走
if (p2->right == nullptr) {
p2->right = p1;
p1 = p1->left;
continue;
// // p2->right == p1 说明该节点已经访问过,此时再访问到就置为空
} else {
p2->right = nullptr;
// 打印p1->left而非p1,就保证根节点一定是最后打印的
addPath(res, p1->left);
}
}
// 1、其左孩子已经访问完毕,因此把p1的右孩子赋给p1
// 要么就是在回溯的路上,要么就是到了一个全新的未被访问过的节点
p1 = p1->right;
}
// 最后打印root及其右孩子(若右孩子是叶子节点)
addPath(res, root);
return res;
}
};
Accepted
- 68/68 cases passed (4 ms)
- Your runtime beats 40.93 % of cpp submissions
- Your memory usage beats 95.58 % of cpp submissions (8 MB)