给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
解题思路:
二叉树的后序遍历,最简单的思想就是递归算法,三步走就可以轻松解决,代码如下:
/**
* 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;
}
dfs(root, res);
return res;
}
void dfs(TreeNode* root, vector<int>& res){
//先向左
if(root->left){
dfs(root->left, res);
}
//再向右
if(root->right){
dfs(root->right, res);
}
//最后才是当前节点
res.push_back(root->val);
}
};
当然还有别的方法,比如迭代的方法,用栈的方式实现,因为是后续遍历,如果当前栈顶元素有右节点,则不pop,root=tmp->right,然后把栈顶元素的右指针置为nullptr。代码如下:
/**
* 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> ans;
stack<TreeNode*> s;
while(root != nullptr || !s.empty()){
if(root != nullptr){
s.push(root);
root = root->left;
}else{
TreeNode* tmp = s.top();
if(tmp->right == nullptr){
s.pop();
ans.push_back(tmp->val);
continue;
}
root = tmp->right;
tmp->right = nullptr;
}
}
return ans;
}
};