C++【迭代法实现前中后序遍历】

前言

二叉树的遍历是一个经典问题,原本只是为了实现对二叉树的遍历,使用递归方法是最易读写和记忆的遍历方式,但是,但递归算法也有一个缺点:由递归调用自身所产生的函数调用栈找不方便,可能会使系统栈溢出,而且递归需要消耗过多的运行时间和系统栈空间,影响程序的效率。因此,如何使用迭代方法对二叉树进行遍历成了一种重要的算法。

一、定义二叉树结点

struct TreeNode {
    
    
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {
    
    }
};

先定义一个结点,里面包含了结点的值,和左右子树指针,以及初始化初始化构造函数。

二、迭代法实现前序遍历

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

在二叉树的遍历过程中,迭代法的基本思路是借助数据结构(如栈、队列等)来保存待访问的节点信息,以达到遍历的目的。
具体操作方法如下:
对于前序遍历,先将根节点入栈,然后重复以下操作:
1.弹出栈顶元素,并将其访问;
2.先将右子节点入栈;
3.再将左子节点入栈。

三、迭代法中序遍历

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

对于中序遍历,先将根节点入栈,然后重复以下操作:
1.若当前节点非空,则将其入栈,并将其左子节点设为当前节点;
2.若当前节点为空,则从栈中弹出一个元素,并访问该元素;
3.将当前节点设为弹出元素的右子节点。

四、迭代法后序遍历

class Solution {
    
    
public:
    vector<int> postorderTraversal(TreeNode* root) {
    
    
         vector<int> res;
        stack<TreeNode*> s;
        TreeNode* pre = nullptr;
        TreeNode* node = root;
        
        while (node || !s.empty())
         {
    
    
            while (node) 
            {
    
    
                s.push(node);
                node = node->left;
            }
        node = s.top();
        if (!node->right || node->right == pre) 
        {
    
    
            res.push_back(node->val);
            s.pop();
            pre = node;
            node = nullptr;
        } 
        else 
        {
    
    
            node = node->right;
        }
    } 
    return res;
}
};

对于后序遍历,先将根节点入栈,然后重复以下操作:
1.若当前节点非空,则将其入栈,并将其左子节点设为当前节点;
2.若当前节点为空,则取出栈顶元素进行判断,若不存在右子节点或者右子节点3.已经被访问过,则将其访问,否则将右子节点设为当前节点。

猜你喜欢

转载自blog.csdn.net/m0_59292239/article/details/130700572