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

本文详细介绍了二叉树的基础概念,如满二叉树、完全二叉树和它们在数据结构中的应用,如优先级队列。此外,还探讨了二叉搜索树、平衡二叉搜索树以及C++中(map,set)底层的实现。重点讲解了前序、中序和后序遍历的递归和非递归方法,以及统一的迭代法解决访问和处理节点的问题。
摘要由CSDN通过智能技术生成

二叉树理论基础

满二叉树

满二叉树:如果一棵二叉树只有度为0的结点和度为2的结点,并且度为0的结点在同一层上(最后一层),则这棵二叉树为满二叉树。

对于深度为k的满二叉树,其结点有2^k-1个。

完全二叉树

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

优先级队列其实是一个堆,堆就是一棵完全二叉树,同时保证父子节点的顺序关系。最后一个二叉树的顺序有误,应该要先满足左边。

 二叉搜索树

二叉树是一个有序树,若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值;若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;它的左、右子树也分别为二叉排序树

平衡二叉搜索树

一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。

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

二叉树可以链式存储,也可以顺序存储。链式存储方式用指针, 顺序存储的方式是用数组。

二叉树遍历

  • 深度优先遍历
    • 前序遍历(递归法,迭代法)中左右
    • 中序遍历(递归法,迭代法)左中右
    • 后序遍历(递归法,迭代法)左右中
  • 广度优先遍历
    • 层次遍历(迭代法)

递归遍历

递归三要素

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

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

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

前序遍历(中左右)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
//确定参数和返回值
    void travelsal(TreeNode*cur, vector<int> &vec){
//确定递归终止条件
        if(cur==nullptr)return;
//确定单层递归的逻辑:前序遍历是中左右的循序,所以在单层递归的逻辑,是要先取中节点的数值,代码如下:
        vec.push_back(cur->val);
        travelsal(cur->left,vec);
        travelsal(cur->right,vec);
    }

    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res;
        travelsal(root,res);
        return res;
    }
};

后序遍历(左右中)


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

中序遍历(左中右)

class Solution {
public:
    void travelval(TreeNode* cur,vector<int> &vec){
        if(cur==nullptr)return;
        travelval(cur->left,vec);
        vec.push_back(cur->val);
        travelval(cur->right,vec);
    }

    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        travelval(root,res);
        return res;
    }
};

非递归遍历(迭代法)

使用递归的方法通常也可以通过栈来实现。

前序

出现错误

忽略了空二叉树的情况,未加入if (root == NULL) return res;

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* node=st.top();
          st.pop();
          res.push_back(node->val);
          if (node->right) st.push(node->right);
          if(node->left) st.push(node->left);
      }
      return res;
    }
};

后序

由于前序是中左右,if设置为中右左。要实现后续,可以将if设置为中左右,得到的是中右左,然后通过reverse之后便可以得到左右中。

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

中序

相比较前序与后续,中序会出现操作结点与遍历结点不一样的情况。在中序遍历时,需要借助指针指向需要操作的结点,用堆栈存储遍历结点。

出现错误

定式思维直接在堆栈中先插入了根结点,然而while循环中的if语句就会实现此效果

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> inorderTraversal(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(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
                if (node->right) st.push(node->right);  // 添加右节点(空节点不入栈)

                st.push(node);                          // 添加中节点
                st.push(NULL); // 中节点访问过,但是还没有处理,加入空节点做为标记。

                if (node->left) st.push(node->left);    // 添加左节点(空节点不入栈)
            } else { // 只有遇到空节点的时候,才将下一个节点放进结果集
                st.pop();           // 将空节点弹出
                node = st.top();    // 重新取出栈中元素
                st.pop();
                result.push_back(node->val); // 加入到结果集
            }
        }
        return result;
    }
};
//前序
class Solution {
public:
    vector<int> preorderTraversal(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();
                if (node->right) st.push(node->right);  // 右
                if (node->left) st.push(node->left);    // 左
                st.push(node);                          // 中
                st.push(NULL);
            } else {
                st.pop();
                node = st.top();
                st.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};
//后序
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;
    }
};

  • 23
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值