题解 | #二叉树的后序遍历#

二叉树的后序遍历

https://www.nowcoder.com/practice/1291064f4d5d4bdeaefbf0dd47d78541

递归

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型vector
     */
     vector<int> ans;
    vector<int> postorderTraversal(TreeNode* root) {
       dfs(root);
        return ans;
    }

    void dfs(TreeNode* root) {
        if (!root) return;
        dfs(root->left);
        dfs(root->right);
        ans.push_back(root->val);
    }
};

迭代

思路:先遍历根右左然后翻转过来变成左右根,也就成了我们的后续遍历

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型vector
     */
    vector<int> postorderTraversal(TreeNode* root) {
        // write code here
        vector<int> res;
        stack<TreeNode*> stk;
        while(root || stk.size()) {
            while(root) {
                res.push_back(root->val);  // 先根
                stk.push(root);  // 把root压入栈
                root = root->right;  // 遍历右边
            }
            root = stk.top()->left;  // 遍历完所有右边就回溯直接遍历左边
            stk.pop();  // 弹出
        }
        // 翻转答案  原先是根右左翻转过来就是左右根这样就变成了后序遍历
        reverse(res.begin(), res.end());  // 翻转答案,把1 2 3 翻转过来就是我们要的答案 3 2 1
        return res;
    }
};
#二叉树的后序遍历#
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务