剑指54二叉搜索树的第K大结点 55 二叉树的深度

最新推荐文章于 2025-05-04 20:55:54 发布

丫丫爱吃

最新推荐文章于 2025-05-04 20:55:54 发布

阅读量140

点赞数

文章标签：深度优先算法

本文链接：https://blog.youkuaiyun.com/meiyongyue/article/details/131944910

版权

文章提供了两种树的遍历方法，一种是使用深度优先搜索（DFS）找到二叉树中的第k个最大值，另一种是利用广度优先搜索（BFS）计算树的最大深度。在DFS中，结果存储并排序后返回，而在BFS中，通过层次遍历计算最大深度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

class Solution {
public:
    int kthLargest(TreeNode* root, int k) {
        dfs(root);
        sort(result.begin(),result.end());
        reverse(result.begin(),result.end());
        if(k>=result.size()) return result[result.size()-1];
        return result[k-1];
    }
private:
   vector<int> result;
   void dfs(TreeNode* root)
   {
       if(!root) return ;
       result.push_back(root->val);
       dfs(root->left);
       dfs(root->right);
   }
};

在这里插入图片描述

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(!root) return 0;
        dfs(root,0);//从0开始计数
        return depth;
    }
private:
    int depth=1;
    void dfs(TreeNode* root,int dep)
    {
        if(!root) 
        {
         depth=max(depth,dep);
         return ;
        }
        dep++;
        dfs(root->left,dep);//dfs前面会对每个进来的结点判空 空就return 所以这边不要判空
        dfs(root->right,dep);
    }
};

采用迭代队列写写看

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(!root) return 0;//不要忘记好吧求求你了
        int depth=0;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
            int size=q.size();
            while(size--)
            {
                TreeNode* t=q.front();
                q.pop();
                if(t->left) q.push(t->left);
                if(t->right) q.push(t->right);          
            }
            depth++;
        }
        return depth;
    }
};