题目链接:145. 二叉树的后序遍历
题目:
给你一棵二叉树的根节点 root ,返回其节点值的 后序 遍历 。
示例 1:
输入:root = [1,null,2,3]
输出:[3,2,1]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
提示:
- 树中节点的数目在范围 [0, 100] 内
- -100 <= Node.val <= 100
进阶:递归算法很简单,你可以通过迭代算法完成吗?
思路和算法:
本题使用递归法会比较容易些,明确递归三要素就可以很快的解决,具体三要素的讲解见:叫你写递归——递归三部曲。
在本题中,三要素是指:
(1)确定递归函数的参数和返回值:因为要打印出后序遍历节点的数值,所以参数里需要传入vector在放节点的数值,除了这一点就不需要在处理什么数据也不需要什么返回值,所以递归函数返回类型可以敲定为void;
即:void traversal(TreeNode* cur, vector<int>& res) {}
(2)确定终止条件:在递归的过程中,什么才能算递归结束了呢,当然是当前遍历的节点是空,那么本层递归就要结束了,所以,如果当前遍历的这个节点是空,直接return;
即:if (!cur) return;
(3)确定单层递归的逻辑:后序遍历是左右中的顺序,所以在单层递归的逻辑,是先递归左右子二叉树,最后再取中节点的数值放入结果数组;即:
traversal(cur->left, res); //左
traversal(cur->right, res); //右
res.push_back(cur->val); //将当前节点放入结果数组
代码(c++):
/**
* 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 {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
traversal(root, res);
return res;
}
private:
void traversal(TreeNode* cur, vector<int>& res) {
if (!cur) return; //如果当前节点为空,则结束递归
traversal(cur->left, res); //左
traversal(cur->right, res); //右
res.push_back(cur->val); //将当前节点放入结果数组
}
};
关于进阶问题采用的迭代法。
代码(c++):
由于前序遍历是中左右,所以我们可以在前序遍历迭代法的基础上进行修改:
(1)首先压栈顺序不再是先将右孩子压栈后将左孩子压栈,而是反过来,先将左孩子压栈而后再将右孩子压栈,这样得到的的结果数组顺序为中右左;
(2)这时将得到的中右左顺序的结果数组反转顺序,就可以得到左右中顺序的结果数组。
/**
* 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 {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res; //结果数组
if (!root) return res;
stack<TreeNode*> stk;
stk.push(root);
//如果栈内有数据
while (!stk.empty()) {
TreeNode* cur = stk.top();
stk.pop();
res.push_back(cur->val); //将栈顶元素放入结果数组
if (cur->left) stk.push(cur->left); //如果左节点不为空,则压栈
if (cur->right) stk.push(cur->right);
}
reverse(res.begin(), res.end()); //反转结果数组
return res;
}
};