LinCode 刷题 之二叉树最小深度

http://www.lintcode.com/zh-cn/problem/minimum-depth-of-binary-tree/  题目描述信息

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */

给出一棵如下的二叉树:

1

/     \

2       3

/    \

4      5

这个二叉树的最小深度为 2

思路 某个节点的左子树和右子数都没有 就是最小深度，广度优先遍历二叉树即可（队列）

public class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: An integer.
     */
    public int minDepth(TreeNode root) {
       Queue<TreeNode> queue=new LinkedList<TreeNode>();
    
        int num;
        if(root==null){
            num=0;
        }else{
            num=1;
            queue.add(root);
        }
        while(!queue.isEmpty()){
            int size = queue.size();
            
            for(int i=0;i<size;i++){
                TreeNode node = queue.poll();
                if(node.left==null&&node.right==null){
                    queue=new LinkedList<>();
                   return num;
                }else{
                    if(node.left!=null){
                        queue.offer(node.left);
                    }
                    
                    if(node.right!=null){
                        queue.offer(node.right);
                    }
                }
            }
            
            num++;
            
        }
        
        return num;
    }
}