给一颗二叉树,找这颗二叉树最小的深度。最小深度是从根节点到最近的叶节点的最短路径上的节点数。
思路://采用广度优先搜索,或者层序遍历,找到的第一个叶节点的深度即是最浅。
一定记得维护一个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;
}
}