leetcode NO.111 二叉树的最小深度（递归和迭代）

最新推荐文章于 2024-11-09 05:20:46 发布

Koma_Kong

最新推荐文章于 2024-11-09 05:20:46 发布

阅读量89

点赞数 2

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/Koma_Kong/article/details/139483493

版权

递归做法：

用全局变量记录最小的深度，函数内的局部变量记录到达当前节点的深度

class Solution {
public:
    int depth = 1e9;
    int min(int a, int b) { return a < b ? a : b; }
    void search(TreeNode* root, int count) {
        if (root == nullptr)
            return;
        count++;
        if (root->left == nullptr && root->right == nullptr)
            depth = min(depth, count);
        search(root->left, count);
        search(root->right, count);
    }
    int minDepth(TreeNode* root) {
        if (root == nullptr)
            return 0;
        search(root, 0);
        return depth;
    }
};

迭代法：

层序遍历到每一层时，如有叶子节点则返回当前层数

class Solution {
public:
    int minDepth(TreeNode* root) {
queue<TreeNode*> w;
        if (root)
            w.push(root);
        int minDepth = 0;
        while (!w.empty()) {
            TreeNode* node;
            int size = w.size();
            minDepth++;
            for (int i = 0; i < size; i++) {
                node = w.front();
                if (node->left == nullptr && node->right == nullptr) 
                        return minDepth ;
                if (node->left)
                    w.push(node->left);
                if (node->right)
                    w.push(node->right);
                w.pop();
            }
        }
        return minDepth;
    }
};