首页 > 编程语言 >算法训练 | 二叉树Part2 | 层序遍历、226.翻转二叉树、101.对称二叉树

算法训练 | 二叉树Part2 | 层序遍历、226.翻转二叉树、101.对称二叉树

时间:2024-05-27 16:31:37浏览次数:28  
标签:right return 层序 Part2 二叉树 TreeNode root 节点 left

目录

广度优先

226.翻转二叉树

递归法 ⭐

迭代法

层序法

101.对称二叉树

后序遍历法 ⭐

迭代法


嵌入式学习分享个人主页:Orion嵌入式随想录 - 小红书 (xiaohongshu.com)

广度优先
  • 解题思路

    • 层序遍历一个二叉树。就是从左到右一层一层的去遍历二叉树。

    • 需要借用一个辅助数据结构即队列来实现,队列先进先出,符合一层一层遍历的逻辑,而用栈先进后出适合模拟深度优先遍历也就是递归的逻辑。这种层序遍历方式就是图论中的广度优先遍历,应用在二叉树上。

  • 解题步骤

    • 定义队列,增加节点

    • 定义结果数组

    • 开启循环,定义长度和存放数组,队列不为空则按照队列长度存储遍历节点

    • 存入一组结果

  • 代码注意

    • root != NULL放入头节点

    • vector<int> vec;在for前定义

  • 代码一:层序遍历

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            // 这里一定要使用固定大小size,不要使用que.size(),因为que.size是不断变化的
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(vec);
        }
        return result;
    }
};

// 递归法
class Solution {
public:
    void order(TreeNode* cur, vector<vector<int>>& result, int depth)
    {
        if (cur == nullptr) return;
        if (result.size() == depth) result.push_back(vector<int>());
        result[depth].push_back(cur->val);
        order(cur->left, result, depth + 1);
        order(cur->right, result, depth + 1);
    }
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> result;
        int depth = 0;
        order(root, result, depth);
        return result;
    }
};

226.翻转二叉树

  • 注意只要把每一个节点的左右孩子翻转一下,就可以达到整体翻转的效果。

  • 这道题目使用前序遍历和后序遍历都可以,唯独中序遍历不方便,因为中序遍历会把某些节点的左右孩子翻转了两次。

  • 层序遍历依然可以的,只要把每一个节点的左右孩子翻转一下的遍历方式都是可以的。

递归法
  • 解题思路

    • 确定递归函数的参数和返回值:参数就是要传入节点的指针,不需要其他参数了,通常此时定下来主要参数,如果在写递归的逻辑中发现还需要其他参数的时候,随时补充。返回值的话其实也不需要,但是题目中给出的要返回root节点的指针,可以直接使用题目定义好的函数,所以就函数的返回类型为TreeNode*

    • TreeNode* invertTree(TreeNode* root)
    • 确定终止条件:当前节点为空的时候,就返回。

    • if (root == NULL) return root;
    • 确定单层递归的逻辑:先前序遍历,所以先进行交换左右孩子节点,然后反转左子树,反转右子树。

    • swap(root->left, root->right);
      invertTree(root->left);
      invertTree(root->right);
  • 代码一:前序遍历递归

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) return root;
        swap(root->left, root->right);  // 中
        invertTree(root->left);         // 左
        invertTree(root->right);        // 右
        return root;
    }
};
  • 代码二:后序遍历递归

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) return root;
        invertTree(root->left);         // 左
        invertTree(root->right);        // 右
        swap(root->left, root->right);  // 中
        return root;
    }
};
  • 代码三:中序遍历递归

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) return root;
        invertTree(root->left);         // 左
        swap(root->left, root->right);  // 中
        invertTree(root->left);        // 右 不是严格的右
        return root;
    }
};
迭代法
  • 代码一:前序遍历迭代

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) return root;
        stack<TreeNode*> st;
        st.push(root);
        while(!st.empty()) {
            TreeNode* node = st.top();              // 中
            st.pop();
            swap(node->left, node->right);
            if(node->right) st.push(node->right);   // 右
            if(node->left) st.push(node->left);     // 左
        }
        return root;
    }
};
  • 代码二:前序统一迭代

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        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();
                swap(node->left, node->right);          // 节点处理逻辑
            }
        }
        return root;
    }
};
层序法
  • 代码一:层序遍历

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                swap(node->left, node->right); // 节点处理
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return root;
    }
};

101.对称二叉树

后序遍历法
  • 解题思路

    • 对于二叉树是否对称,要比较的是根节点的左子树与右子树是不是相互翻转的,理解这一点就知道了其实我们要比较的是两个树(这两个树是根节点的左右子树),所以在递归遍历的过程中,也是要同时遍历两棵树。

    • 比较:比较的是两个子树的里侧和外侧的元素是否相等。

    • 遍历:本题遍历只能是“后序遍历”,因为我们要通过递归函数的返回值(处理完左右返回中处理)来判断两个子树的内侧节点和外侧节点是否相等。正是因为要遍历两棵树而且要比较内侧和外侧节点,所以准确的来说是一个树的遍历顺序是左右中,一个树的遍历顺序是右左中。但都可以理解算是后序遍历,尽管已经不是严格上在一个树上进行遍历的后序遍历了。

  • 解题步骤

    • 确定递归函数的参数和返回值:因为我们要比较的是根节点的两个子树是否是相互翻转的,进而判断这个树是不是对称树,所以要比较的是两个树,参数是左子树节点和右子树节点。返回值是bool类型。

    • 确定终止条件:要比较两个节点数值相不相同,首先要把两个节点为空的情况弄清楚!否则后面比较数值的时候就会操作空指针了。
      • 节点为空的情况有:(注意比较的其实不是左孩子和右孩子,所以如下我称之为左节点右节点)

      • 左右都为空,对称,返回true

      • 左不为空,右为空,不对称 return false

      • 左节点为空,右节点不为空,不对称,return false

      • 此时已经排除掉了节点为空的情况,那么剩下的就是左右节点不为空:

        • 左右都不为空,比较节点数值,不相同就return false

      • 剩下的就是 左右节点都不为空,且数值相同的情况。进入下一步骤。

    • 确定单层递归的逻辑:此时才进入单层递归的逻辑,单层递归的逻辑就是处理 左右节点都不为空,且数值相同的情况。

      • 比较二叉树外侧是否对称:传入的是左节点的左孩子,右节点的右孩子。

      • 比较内侧是否对称,传入左节点的右孩子,右节点的左孩子。

      • 如果左右都对称就返回true ,有一侧不对称就返回false 。

  • 代码一:后序递归

class Solution {
public:
    bool compare(TreeNode* left, TreeNode* right) {
        // 首先排除空节点的情况
        if (left == NULL && right != NULL) return false;
        else if (left != NULL && right == NULL) return false;
        else if (left == NULL && right == NULL) return true;
        // 排除了空节点,再排除数值不相同的情况
        else if (left->val != right->val) return false;

        // 此时就是:左右节点都不为空,且数值相同的情况
        // 此时才做递归,做下一层的判断
        bool outside = compare(left->left, right->right);   // 左子树:左、 右子树:右
        bool inside = compare(left->right, right->left);    // 左子树:右、 右子树:左
        bool isSame = outside && inside;                    // 左子树:中、 右子树:中 (逻辑处理)
        return isSame;
    }
    bool isSymmetric(TreeNode* root) {
        if (root == NULL) return true;
        return compare(root->left, root->right);
    }
};
  • 代码二:后序递归简洁版

class Solution {
public:
    bool compare(TreeNode* left, TreeNode* right) {
        if (left == NULL && right != NULL) return false;
        else if (left != NULL && right == NULL) return false;
        else if (left == NULL && right == NULL) return true;
        else if (left->val != right->val) return false;
        else return compare(left->left, right->right) && compare(left->right, right->left);

    }
    bool isSymmetric(TreeNode* root) {
        if (root == NULL) return true;
        return compare(root->left, root->right);
    }
};
迭代法
  • 解题思路

    • 这道题目也可以使用迭代法,但要注意,这里的迭代法不是前中后序的迭代写法,因为本题的本质是判断两个树是否是相互翻转的,其实已经不是所谓二叉树遍历的前中后序的关系了。

    • 这里我们可以使用队列来比较两个树(根节点的左右子树)是否相互翻转,(注意这不是层序遍历)

  • 代码一:迭代法队列

class Solution {
public:
    bool isSymmetric(TreeNode* root) {
        if (root == NULL) return true;
        queue<TreeNode*> que;
        que.push(root->left);   // 将左子树头结点加入队列
        que.push(root->right);  // 将右子树头结点加入队列
        
        while (!que.empty()) {  // 接下来就要判断这两个树是否相互翻转
            TreeNode* leftNode = que.front(); que.pop();
            TreeNode* rightNode = que.front(); que.pop();
            if (!leftNode && !rightNode) {  // 左节点为空、右节点为空,此时说明是对称的
                continue;
            }

            // 左右一个节点不为空,或者都不为空但数值不相同,返回false
            if ((!leftNode || !rightNode || (leftNode->val != rightNode->val))) {
                return false;
            }
            que.push(leftNode->left);   // 加入左节点左孩子
            que.push(rightNode->right); // 加入右节点右孩子
            que.push(leftNode->right);  // 加入左节点右孩子
            que.push(rightNode->left);  // 加入右节点左孩子
        }
        return true;
    }
};
  • 代码二:迭代法栈:

    • 迭代法,其实是把左右两个子树要比较的元素顺序放进一个容器,然后成对成对的取出来进行比较,那么其实使用栈也是可以的。只要把队列原封不动的改成栈就可以。

class Solution {
public:
    bool isSymmetric(TreeNode* root) {
        if (root == NULL) return true;
        stack<TreeNode*> st; // 这里改成了栈
        st.push(root->left);
        st.push(root->right);
        while (!st.empty()) {
            TreeNode* leftNode = st.top(); st.pop();
            TreeNode* rightNode = st.top(); st.pop();
            if (!leftNode && !rightNode) {
                continue;
            }
            if ((!leftNode || !rightNode || (leftNode->val != rightNode->val))) {
                return false;
            }
            st.push(leftNode->left);
            st.push(rightNode->right);
            st.push(leftNode->right);
            st.push(rightNode->left);
        }
        return true;
    }
};

(说明:基于代码随想录课程学习,部分内容引用代码随想录文章)

标签:right,return,层序,Part2,二叉树,TreeNode,root,节点,left
From: https://blog.csdn.net/qq_48896570/article/details/139125732

相关文章

  • 104.二叉树的最大深度
    给定一个二叉树,找出其最大深度。二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。说明:叶子节点是指没有子节点的节点。示例:给定二叉树[3,9,20,null,null,15,7],返回它的最大深度3。代码:(递归法)#Definitionforabinarytreenode.#classTreeNode:......
  • 链式二叉树的前,中,后序遍历 AND 结点个数及高度等 文末附带全部代码
    目录前言1.前序遍历2.中序遍历3.后续遍历4.二叉树结点的个数5.二叉树叶子结点个数6.二叉树的高度7.二叉树第K层结点的个数8.二叉树查找值为x的结点全部代码总结正文开始前言本文旨在介绍二叉树的链式存储中一些函数的实现博客主页:酷酷学!!!更多文章,......
  • 二叉树遍历算法与堆数据结构详解(C语言)
    目录树的概念及结构二叉树的概念及结构概念二叉树的性质满二叉树和完全二叉树满二叉树完全二叉树深度的计算二叉树顺序结构及实现顺序存储堆的概念数组建堆向下调整堆的实现完整代码Heap.hHeap.cTest.c堆的初始化(实现小堆为例)插入数据删除堆顶的数据 ......
  • 二叉树的基本功能(代码实现)
    1.二叉树的概念在对树的介绍一文中,有明确的介绍。如有兴趣可移步至:数据结构:树的介绍-CSDN博客2.二叉树的代码实现2.1二叉树的结构体typedefintBTDataType;typedefstructBinaryTreeNode{ BTDataTypedata; structBinaryTreeNode*left; structBinaryTreeNo......
  • 代码随想录算法训练营第第18天 | 513.找树左下角的值 、112. 路径总和 、106.从中
    找树左下角的值本地递归偏难,反而迭代简单属于模板题,两种方法掌握一下题目链接/文章讲解/视频讲解:https://programmercarl.com/0513.找树左下角的值.html/***Definitionforabinarytreenode.*functionTreeNode(val,left,right){*this.val=(val===undef......
  • 数据结构:二叉树与树
    一树的基本概念:1.树的形状:2.树的定义:树是一种非线性的数据结构,它是n(n>=0)个结点的有限集。当n=0时,称为空树。在任意一棵非空树中应满足:2.1有且仅有一个特定的称为根的结点。2.2当n>1时,其余结点可分为m(m>0)个互不相交的有限集T1……Tm,其中每个集合本身又是......
  • Day36 代码随想录打卡|二叉树篇---翻转二叉树
    题目(leecodeT226):给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。方法:迭代法翻转二叉树,即从根节点开始,一一交换每个节点的左右孩子节点,然后递归此过程,将根节点的左右孩子节点再分别作为参数传入交换节点的函数中。重复此过程,直到结束。就完成了二叉树的翻......
  • 二叉树翻转
    给定一棵二叉树的根节点 root,请左右翻转这棵二叉树,并返回其根节点。示例1:输入:root=[5,7,9,8,3,2,4]输出:[5,9,7,4,2,3,8]提示:树中节点数目范围在 [0,100] 内-100<=Node.val<=100思路:将二叉树看成一个只有左右子树的最小二叉树,那么翻转这个二叉树就是将左......
  • 代码随想录算法训练营第三十七天|435. 无重叠区间、763.划分字母区间、56. 合并区间、
    435.无重叠区间文档讲解:代码随想录题目链接:.-力扣(LeetCode)本道题与上个题目相似,都是求重叠区间统计重叠区间的个数,减去重叠区间的个数就是无重叠区间了主要就是为了让区间尽可能的重叠。(为什么)按照左边界排序①如果i的左边界大于等于上一个区间的右边界,就没有重叠......
  • 101. 对称二叉树
    给定一个二叉树,检查它是否是镜像对称的。思路:判断左子树和右子树是否可以相互翻转。外侧与外侧比较,内侧与内侧比较。使用哪种遍历方式?这道题只能使用后序(左右中)。因为我们要不断收集左右孩子的信息,返回给上一个节点。然后判断以根节点的左孩子为根节点的信息,和根......