Leetcode:给一颗二叉树,找这颗二叉树最小的深度。

版权声明:本博客为记录本人学习过程而开,内容大多从网上学习与整理所得,若侵权请告知! https://blog.csdn.net/Fly_as_tadpole/article/details/86535871

给一颗二叉树,找这颗二叉树最小的深度。最小深度是从根节点到最近的叶节点的最短路径上的节点数。

思路://采用广度优先搜索,或者层序遍历,找到的第一个叶节点的深度即是最浅。

一定记得维护一个last指针,来指向每一行的最后一个元素!!


C++ 版本

class Solution {
public:
    typedef  TreeNode* tree;
    int run(TreeNode *root) {
        
        if(!root) return 0;
        queue<tree> q;
        tree now,last;
        int level,size;
        last = now = root;
        level = 1;
        q.push(root);
        while(q.size()){
            now = q.front();
            q.pop();
            size = q.size();
            if(now->left) q.push(now->left);
            if(now->right) q.push(now->right);
            //如果走到了每一行的最右边,则将last更新为当前最右节点,也即是last表示每行的行尾;
            //每次push新的节点之后,都要判断一下当前队列的元素数量是否更新;
            //如果没有更新且当前队列为空,则跳出循环;
            if(q.size()-size == 0) break;
            if(last == now){
                level++;
                if(q.size()) last = q.back();
            }
        }
        return level;
    }
};

Java版本

来个非递归的,思路是层序遍历,找到第一个左右结点都为null的情况,就返回.

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.LinkedList;
import java.util.Queue;
/*给定一个二叉树,求其最小深度。最小深度是从根节点到最近的叶节点的最短路径上的节点数。
*/
public class Solution {
    public int run(TreeNode root) {
        if(root == null) return 0;
        if(root.left == null && root.right == null) return 1;
        int depth = 0;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            int len = queue.size();
            depth++;
            for(int i = 0;i<len;i++){
                //poll()返回并删除队头元素
                TreeNode cur = queue.poll();
                if(cur.left == null && cur.right == null) return depth;
                if(cur.left != null) queue.offer(cur.left);
                if(cur.right != null) queue.offer(cur.right);
                }
            }
        return 0;
        }
    }

猜你喜欢

转载自blog.csdn.net/Fly_as_tadpole/article/details/86535871