代码随想录算法训练营第十三天 | 二叉树理论基础、递归遍历、迭代遍历、统一迭代

于 2024-01-04 12:49:23 发布
阅读量918 收藏 24
点赞数 21
分类专栏： 代码随想录 文章标签： 算法 leetcode c++
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Cautial/article/details/134454635
版权
本文详细介绍了二叉树的基础理论，包括满二叉树和完全二叉树的概念，以及它们与优先级队列的关系。探讨了二叉搜索树、平衡二叉搜索树（AVL树）的特点，并比较了链式和顺序存储方式。文章还涵盖了各种遍历方法，如深度优先、前序、中序、后序和广度优先，以及递归和迭代的实现策略。
摘要由CSDN通过智能技术生成

二叉树题目分类：

二叉树理论基础

二叉树种类：满二叉树和完全二叉树

满二叉树

如果一棵二叉树只有度为0的结点和度为2的结点，并且度为0的结点在同一层上，则这棵二叉树为满二叉树。也可以说是深度为k，有2^(k-1)个节点的二叉树。

完全二叉树

在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层（h从1开始），则该层包含 1~ 2^(h-1) 个节点。

优先级队列其实是一个堆，堆就是一棵完全二叉树，同时保证父子节点的顺序关系。

二叉搜索树

二叉搜索树是一个有序树。

  • 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值；
  • 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；
  • 它的左、右子树也分别为二叉排序树

平衡二叉搜索树

又被称为AVL（Adelson-Velsky and Landis）树，且具有以下性质：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

C++中map、set、multimap，multiset的底层实现都是平衡二叉搜索树，所以map、set的增删操作时间时间复杂度是logn，注意我这里没有说unordered_map、unordered_set，因为他们的底层实现是哈希表。

二叉树的存储方式

二叉树可以用指针链式存储，这种存储方式的元素在内存中就不是连续分布的；也可以用数组进行顺序存储，这种存储方式的元素在内存中就是连续分布的。

链式存储：

顺序存储： 

如果父节点的数组下标是 i，那么它的左孩子就是 i * 2 + 1，右孩子就是 i * 2 + 2。

二叉树的遍历方式

图论中最基本的两种遍历方式：

  • 深度优先遍历：先往深走，遇到叶子节点再往回走。
    • 前序遍历（递归法，迭代法）
    • 中序遍历（递归法，迭代法）
    • 后序遍历（递归法，迭代法）
  • 广度优先遍历：一层一层的去遍历。
    • 层次遍历（迭代法）

深度优先遍历的前中后，其实指的就是中间节点的遍历顺序

  • 前序遍历：中左右
  • 中序遍历：左中右
  • 后序遍历：左右中

二叉树相关题目里，经常会使用递归的方式来实现深度优先遍历。之前我们讲栈与队列的时候，就说过栈其实就是递归的一种实现结构，也就说前中后序遍历的逻辑其实都是可以借助栈使用递归的方式来实现的。

广度优先遍历的实现一般使用队列来实现，需要先进先出的结构才能一层一层的来遍历二叉树。

二叉树的定义

顺序存储就是用数组来存。

链式存储的二叉树节点的定义方式：

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

二叉树的定义和链表是差不多的，只是二叉树的节点里有两个指针，分别指向左右孩子。

问题

完全二叉树和平衡二叉搜索树的区别：完全二叉树最后一排的节点是从左向右依次排开的，平衡二叉搜索树最后一排的节点并不一定从左边排开；平衡二叉搜索树是一种二叉搜索树，每个节点都有对应值，且有序排列。

二叉树的递归遍历

递归算法的三个要素

  1. 确定递归函数的参数和返回值： 确定哪些参数是递归的过程中需要处理的，那么就在递归函数里加上这个参数， 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。

  2. 确定终止条件： 写完了递归算法, 运行的时候，经常会遇到栈溢出的错误，就是没写终止条件或者终止条件写的不对，操作系统也是用一个栈的结构来保存每一层递归的信息，如果递归没有终止，操作系统的内存栈必然就会溢出。

  3. 确定单层递归的逻辑： 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。

代码实现

以前序遍历为例，按着如上三步骤确定代码

  1. 确定递归函数的参数和返回值 
    void traversal(TreeNode* cur, vector<int>& vec)
  2.  确定终止条件 
    if (cur == NULL) return;
  3. 确定单层递归逻辑 
    vec.push_back(cur->val);     //中
traversal(cur->left, vec);   //左
traversal(cur->right, vec);  //右

 合起来形成完整的前序遍历代码：

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> res;
        traversal(root, res);
        return res;
    }
};

同理可得中序遍历和后序遍历

中序遍历

void traversal(TreeNode* cur, vector<int>& vec) {
    if (cur == NULL) return;
    traversal(cur->left, vec);
    vec.push_back(cur->val);
    traversal(cur->right, vec);
}

后序遍历

void traversal(TreeNode* cur, vector<int>& vec) {
    if (cur == NULL) return;
    traversal(cur->left, vec);
    traversal(cur->right, vec);
    vec.push_back(cur->val);
}

二叉树的迭代遍历

前序遍历

其实迭代遍历法也是定义一个栈，把二叉树中的节点插入到栈中再取出来实现的，递归法则是直接利用递归函数实现了调用栈的过程。

要注意的是，迭代遍历法中因为栈的先进后出原则，插入栈时要先插入右节点，再插入左节点，这样从栈中取出的时候就会先出左节点再出右节点。

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

中序遍历

中序遍历的顺序要从左边的节点开始，经过中间节点，最后遍历右侧节点。但输入时是从根节点开始访问的，因此上述的前序遍历代码无法简单修改后直接用到中序遍历。

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

后序遍历

前序遍历的遍历顺序是中左右，在往栈中插入时是先插入的右节点再插入左节点，保证出栈的顺序是先左后右，前序遍历的翻转过来就是右左中。

后序遍历可以利用翻转，只是在入栈时先插入左节点再插入右节点，这样出栈后得到的数组就是中右左，翻转一下就得到左右中的顺序。

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

二叉树的统一迭代法

前面说到的迭代法并不能做到统一格式，因为只有前序遍历的访问顺序和处理顺序是一致的。

统一迭代法中就把中间节点做个空指针标记，当再遇到的时候再进行处理。

中序遍历：

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

前序遍历：

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

后序遍历：

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

心得：

第一次学习二叉树写法，反复看了好多遍才记住，统一迭代法代码工整，需要多加复习。

Cautial
关注
  • 21
    点赞
  • 踩
  • 24
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值