统一迭代法

Aurora2005

于 2024-08-29 12:11:08 发布

阅读量170

点赞数 6

分类专栏：数据结构 leetcode 代码随想录文章标签：算法

本文链接：https://blog.csdn.net/2301_80161204/article/details/141675011

版权

数据结构同时被 3 个专栏收录

95 篇文章 0 订阅

订阅专栏

leetcode

93 篇文章 0 订阅

订阅专栏

代码随想录

54 篇文章 0 订阅

订阅专栏

总体思路：

94. Binary Tree Inorder Traversal中序遍历

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int>result;
        stack<TreeNode*>stk;
        if(root!=NULL)stk.push(root);
        while(!stk.empty()){
            TreeNode*node=stk.top();
            if(node!=NULL){
                stk.pop();
                if(node->right)stk.push(node->right);
                stk.push(node);
                stk.push(NULL);
                if(node->left)stk.push(node->left);
            }else{
                stk.pop();
                node=stk.top();
                stk.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};

144. Binary Tree Preorder Traversal

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int>result;
        stack<TreeNode*>stk;
        if(root!=NULL)stk.push(root);
        while(!stk.empty()){
            TreeNode*node=stk.top();
            if(node!=NULL){
                stk.pop();
                if(node->right)stk.push(node->right);
                if(node->left)stk.push(node->left);
                stk.push(node);
                stk.push(NULL);
            }else{
                stk.pop();
                node=stk.top();
                stk.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};

145. Binary Tree Postorder Traversal

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != NULL) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node != NULL) {
                st.pop();
                st.push(node);                          // 中
                st.push(NULL);

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

            } else {
                st.pop();
                node = st.top();
                st.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};