非递归方式实现树的前序后序遍历

最新推荐文章于 2024-10-08 20:32:22 发布

小曾同学杀疯了

最新推荐文章于 2024-10-08 20:32:22 发布

阅读量224

点赞数 1

文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/zengkui198513/article/details/141114514

版权

谦虚
class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> stack;
        vector<int> result;
        stack.push(root);
        while (!stack.empty()) {
            TreeNode* node = stack.top();
            stack.pop();
            if (node == nullptr) {
                continue;
            } else {
                result.push_back(node->val); // 中
                stack.push(node->right); // 右
                stack.push(node->left); // 左      
            }
        }
        reverse(result.begin(),result.end());
        return result;
    }
};

细节：前序遍历是中左右后续遍历时左右中
这里就是先把这两行代码替换一下

           stack.push(node->right); // 右
           stack.push(node->left); // 左

然后整体的结果不就是中右左了嘛，然后这时候对其进行反转，结果如何呢？左右中卧槽，帧厉害

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> stack;
        vector<int> result;
        stack.push(root);
        while (!stack.empty()) {
            TreeNode* node = stack.top();
            stack.pop();
            if (node == nullptr) {
                continue;
            } else {
                result.push_back(node->val);
                stack.push(node->left);
                stack.push(node->right);
                
            }
        }
        reverse(result.begin(),result.end());
        return result;
    }
};