二叉树遍历总结 -- 基于LeetCode

前序遍历 :

144 . 前序遍历

递归

就是按照中前后的顺序依次访问就好了 ;

复制代码
class Solution {
public:
    void traversal(TreeNode* cur , vector<int>& vec){
        if(cur == NULL) return ;
        vec.push_back(cur->val) ;
        traversal(cur->left , vec) ;
        traversal(cur->right , vec) ;
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> ans ;
        traversal(root , ans) ;
        return ans ;
    }
};

迭代

前序遍历是中左右,每次先处理的是中间节点,那么先将根节点放入栈中,然后将右孩子加入栈,再加入左孩子。

为什么要先加入 右孩子,再加入左孩子呢? 因为这样出栈的时候才是中左右的顺序。

复制代码
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> st ;
        vector<int> ans ;
        // 前序 : 中左右 
        // 每次先处理的是中间节点,那么
        // 先将根节点放入栈中,然后将右孩子加入栈,再加入左孩子。
        if(root == NULL) return ans ;
        st.push(root);
        while(!st.empty()){
            TreeNode* cur = st.top() ;
            st.pop() ;
            ans.push_back(cur->val);
            if(cur->right) st.push(cur->right) ;
            if(cur->left) st.push(cur->left) ;
        } 
        return ans ;
    }
};

中序遍历

94 . 二叉树的中序遍历

递归

按照左中右的顺序递归即可 ;

复制代码
class Solution {
public:
    void transfer(TreeNode* cur , vector<int>& vec){
        if(cur == nullptr) return;
        transfer(cur->left,vec);
        vec.push_back(cur->val);
        transfer(cur->right,vec);
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        transfer(root,res);
        return res;
    }
};

迭代

因为是左中右 ,这里需要借助cur指针来帮助迭代 ;

复制代码
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while (cur != NULL || !st.empty()) {
            if (cur != NULL) { // 指针来访问节点,访问到最底层
                st.push(cur); // 将访问的节点放进栈
                cur = cur->left;                // 左
            } else {
                cur = st.top(); // 从栈里弹出的数据,就是要处理的数据(放进result数组里的数据)
                st.pop();
                result.push_back(cur->val);     // 中
                cur = cur->right;               // 右
            }
        }
        return result;
    }
};

后续遍历

145 . 二叉树的后续遍历

递归

左右中

复制代码
class Solution {
public:
    void transfer(TreeNode* cur,vector<int>& vec){
        if(cur == nullptr) return;
        transfer(cur->left,vec);
        transfer(cur->right,vec);
        vec.push_back(cur->val);
    }
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        transfer(root,res);
        return res;
    }
};

迭代

左右中 , 可以先转换成中右左 , 然后最后将ans反转一下即可 ;

复制代码
class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> result;
        if (root == NULL) return result;
        st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            st.pop();
            result.push_back(node->val);
            if (node->left) st.push(node->left); // 相对于前序遍历,这更改一下入栈顺序 (空节点不入栈)
            if (node->right) st.push(node->right); // 空节点不入栈
        }
        reverse(result.begin(), result.end()); // 将结果反转之后就是左右中的顺序了
        return result;
    }
};

二叉树的层序遍历

以下是例题

102. 二叉树的层序遍历

以102为例 :

层序遍历就是从上到下,从左往右依次访问结点 ;

这里借助queue队列这种树结构就可以实现 :

先将根节点放入队列中,然后依次队列中的所有结点,再把其左右结点放入队列中,为下一层的遍历作准备 ;

复制代码
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        vector<vector<int>> ans;
        if(root != nullptr) que.push(root);
        while(!que.empty()){
            int size = que.size();
            vector<int> tmp;
            for(int i=0;i<size;i++){
                TreeNode* node = que.front();
                que.pop();
                tmp.push_back(node->val);
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            ans.push_back(tmp);
        }
        return ans;
    }
};
相关推荐
SweetCode8 分钟前
裴蜀定理:整数解的奥秘
数据结构·python·线性代数·算法·机器学习
ゞ 正在缓冲99%…21 分钟前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xuanjiong22 分钟前
纯个人整理,蓝桥杯使用的算法模板day2(0-1背包问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·动态规划
惊鸿.Jh41 分钟前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
明灯L42 分钟前
《函数基础与内存机制深度剖析:从 return 语句到各类经典编程题详解》
经验分享·python·算法·链表·经典例题
碳基学AI1 小时前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义免费下载方法
大数据·人工智能·python·gpt·算法·语言模型·集成学习
补三补四1 小时前
机器学习-聚类分析算法
人工智能·深度学习·算法·机器学习
独好紫罗兰1 小时前
洛谷题单3-P5718 【深基4.例2】找最小值-python-流程图重构
开发语言·python·算法
正脉科工 CAE仿真1 小时前
基于ANSYS 概率设计和APDL编程的结构可靠性设计分析
人工智能·python·算法
Dovis(誓平步青云)2 小时前
【数据结构】排序算法(中篇)·处理大数据的精妙
c语言·数据结构·算法·排序算法·学习方法