力扣:145. 二叉树的后序遍历

题目链接:145. 二叉树的后序遍历
题目

给你一棵二叉树的根节点 root ,返回其节点值的 后序 遍历 。

示例 1:
力扣:145. 二叉树的后序遍历_第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& 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;
    }
};

你可能感兴趣的:(#,二叉树,leetcode,深度优先,c++,递归,迭代)