• 二叉树前序、中序、后序非递归遍历 144. Binary Tree Preorder Traversal 、 94. Binary Tree Inorder Traversal 、145. Binary Tree Postorder Traversal 、173. Binary Search Tree Iterator


    144. Binary Tree Preorder Traversal

    前序的非递归遍历:用堆来实现

    如果把这个代码改成先向堆存储左节点再存储右节点,就变成了每一行从右向左打印

    如果用队列替代堆,并且先存储左节点,再存储右节点,就变成了逐行打印

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

    94. Binary Tree Inorder Traversal

    思路:用一个变量node记录当前节点,每次先遍历并存储所有的左节点直到为空,然后栈里顶部存储的那个节点就是最近父节点,然后再去遍历一个右节点,在右节点中继续寻找左节点

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

    145. Binary Tree Postorder Traversal

    http://www.cnblogs.com/grandyang/p/4251757.html的第二种方法

    思路:前序遍历是中->左->右,后序遍历是左->右->中。在前序遍历的基础上,insert到begin的方式相当于把顺序全调换了,即变成了右->左->中,这个时候只需要再调换依稀右和左就能变成后序遍历的顺序。

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

    173. Binary Search Tree Iterator

    这个题其实就是中序非递归遍历,并且就是另一种写法的实现,也就是先在循环外将左节点全部装入stack

    注意:初始化只是将第一次所有的左节点装入stack,next是一次一次的递归

    class BSTIterator {
    public:
        BSTIterator(TreeNode* root) {
            while(root){
                sta.push(root);
                root = root->left;
            }
        }
        
        /** @return the next smallest number */
        int next() {
            TreeNode* node = sta.top();
            sta.pop();
            TreeNode* root = node->right;
            while(root){
                sta.push(root);
                root = root->left;
            }
            return node->val;
        }
        
        /** @return whether we have a next smallest number */
        bool hasNext() {
            return !sta.empty();
        }
        stack<TreeNode*> sta;
    };
  • 相关阅读:
    一文了解快排的各种形式和坑
    逆序对的数量
    __type_traits 技法
    Traits 编程技法
    一个简单的内存管理器
    「ZooKeeper」概述
    「MySQL」explain
    「MySQL」InnoDB加锁情况
    「MySQL45讲」个人总结
    「MySQL45讲」1-45讲
  • 原文地址:https://www.cnblogs.com/ymjyqsx/p/9463580.html
Copyright © 2020-2023  润新知