代码随想录算法训练营第11天|二叉树part01

tt555555555555

于 2024-08-21 02:30:00 发布

阅读量65

点赞数 4

分类专栏： C++学习文章标签：算法数据结构 leetcode c++

本文链接：https://blog.csdn.net/zxjiaya/article/details/141361675

版权

C++学习专栏收录该内容

13 篇文章 0 订阅

订阅专栏

226.翻转二叉树

这道题目一些做过的同学理解的也不够深入，建议大家先看我的视频讲解，无论做过没做过，都会有很大收获。

/**
 * 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:
    TreeNode* invertTree(TreeNode* root) {
    if(root==nullptr)
    return root;
    swap(root->left,root->right);
    invertTree( root->left); 
    invertTree(root->right) ;
    return root;
    }
};

题目链接/文章讲解/视频讲解：代码随想录

递归法还是挺简单的，先把自身下面的俩交换了，在让下面的左右各自交换。

/**
 * 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:
    TreeNode* invertTree(TreeNode* root) {
    stack<TreeNode*>st;
    if(root!=nullptr)
    st.push(root);
    while(!st.empty())
    {
        TreeNode* node=st.top();
        st.pop();
        swap(node->left,node->right);
        if(node->left!=nullptr)st.push(node->left);
        if(node->right!=nullptr)st.push(node->right);
    }
    return root;
    }
};

前序遍历法也很简单，没什么难度。经典的，定义栈，交换，压入，弹出。

层序遍历法也不难

/**
 * 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:
    TreeNode* invertTree(TreeNode* root) {
    queue<TreeNode*>que;
    if(root!=nullptr)
    que.push(root);
    while(!que.empty())
    {
        TreeNode* node=que.front();
        que.pop();
        swap(node->left,node->right);
        if(node->left!=nullptr)que.push(node->left);
        if(node->right!=nullptr)que.push(node->right);
    }
    return root;
    }
};

感觉逆转二叉树主要是考虑自己对于二叉树遍历的掌握，整体难度不大，递归法最简单，一直递归。先序遍历用stack，层序遍历用queue

101. 对称二叉树

/**
 * 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:
    bool compare(TreeNode* left,TreeNode* right)
    {
        if(left==nullptr&&right==nullptr)return true;
        else if(left==nullptr&&right!=nullptr)return false;
        else if(left!=nullptr&&right==nullptr)return false;
        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==nullptr){
            return true;
        }
        return compare(root->left,root->right);

    }
};

先看视频讲解，会更容易一些。

还是比较简单的，递归的思想真的好重要，前面几行if语句，本质上就是比较两个指针的值是不是相等，但是比较的最大问题是如果是空指针的话就无法比较。所以还是得考虑如果是空指针怎么办，需要很多if语句

题目链接/文章讲解/视频讲解：代码随想录

/**
 * 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:
   
    bool isSymmetric(TreeNode* root) {
        if(root==nullptr){
            return true;
        }
       queue<TreeNode*>que;  
       que.push(root->left);
       que.push(root->right);     
       while(!que.empty())
       { 
        TreeNode* left=que.front();
       que.pop();
        TreeNode* right=que.front();
        que.pop();
        if(!left&&!right) continue;
        if((!left||!right)||(left->val!=right->val)) return false;
que.push(left->left);
que.push(right->right);
que.push(right->left);
que.push(left->right);
       }
       return true;
    }
};

队列法也挺好理解

堆栈法本质上和队列法完全一样。无论是队列还是堆栈，本质上都是存储两两一对。

/**
 * 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:
   
    bool isSymmetric(TreeNode* root) {
        if(root==nullptr){
            return true;
        }
       stack<TreeNode*>st;  
       st.push(root->left);
       st.push(root->right);     
       while(!st.empty())
       { 
        TreeNode* left=st.top();
       st.pop();
        TreeNode* right=st.top();
       st.pop();
        if(!left&&!right) continue;
        if((!left||!right)||(left->val!=right->val)) return false;
st.push(left->left);
st.push(right->right);
st.push(right->left);
st.push(left->right);
       }
       return true;
    }
};

104.二叉树的最大深度（优先掌握递归）

什么是深度，什么是高度，如何求深度，如何求高度，这里有关系到二叉树的遍历方式。

大家要先看视频讲解，就知道以上我说的内容了，很多录友刷过这道题，但理解的还不够。

题目链接/文章讲解/视频讲解：代码随想录

最先想到的便是层次遍历。层次遍历，每遍历一层，层数++,这样给遍历完。

/**
 * 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:
    int maxDepth(TreeNode* root) {     
        if(!root) return 0;
        queue<TreeNode*>que;
        que.push(root);
       int depth=0;
        while(!que.empty())
        {
            int size=que.size();
            for(int i=0;i<size;i++)
            {
                TreeNode*node=que.front();
                que.pop();
                if(node->left)
                que.push(node->left);
                if(node->right)
                que.push(node->right);     
            }
             depth ++;
        }
        return depth;
        }
};

如果采用递归法，没想到竟然两遍直接通过了

/**
 * 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:
int getDepth(TreeNode* root)
{
    if( root==nullptr )
return 0;
    return max(getDepth(root->left),getDepth(root->right))+1;

}
    int maxDepth(TreeNode* root) {         
        return getDepth(root);
        }
};

就是一个个判断子树的深度。如果子树为空就会返回0，一层层的往上传递。

111.二叉树的最小深度（优先掌握递归）

先看视频讲解，和最大深度看似差不多，其实差距还挺大，有坑。

首先想到的就是层次遍历，遍历到俩子节点都为空，就是多少深度，整个代码本质上就是个深度遍历。

/**
 * 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:
    int minDepth(TreeNode* root) {
if(!root) return 0;
        queue<TreeNode*>que;
        que.push(root);
       int depth=1;
        while(!que.empty())
        {
            int size=que.size();
            for(int i=0;i<size;i++)
            {
                TreeNode*node=que.front();
                que.pop();
               if(!node->left&&!node->right)
               {
                return depth++;
               }
                else{
                    if(node->left)
                    que.push(node->left);
                    if(node->right)
                    que.push(node->right);
                    }
            }
             depth ++;
        }
        return depth;
    
    }
};

然后是递归，也很简单，主要是判断条件，比最大的麻烦多了。因为没有左子树或者右子树并不是最小深度，所以麻烦一些。

/**
 * 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:
int getDepth(TreeNode* node)
{
    if(!node->left&&!node->right)
    return 1;
    if(!node->left&&node->right)
    return 1+getDepth(node->right);
    if(node->left&&!node->right)
    return 1+getDepth(node->left);
    return min(getDepth(node->left),getDepth(node->right))+1;

}
    int minDepth(TreeNode* root) {
        if (!root)
        return 0;
return getDepth(root);
    }
};

题目链接/文章讲解/视频讲解：代码随想录