题目:
给定一个二叉树,在树的最后一行找到最左边的值。
思路:遍历每一层,输出最后一层的节点的值。
如上图:最终的结果应该是 7 。
接下来是应该怎么去设计算法:
- 用一个容器去保存节点,该容器应该满足先进先出(保证优先输出左边)。毋庸置疑就是队列来做。
- 先将根节点放入队列中(第一层放入)。
- 获取队列的大小,如果为空,则返回结果。
- 如果队列不为空,则判断当前节点是否有左右子节点,有则放入队列。
- 如果是第一次访问该节点,则可以认为结果就是该节点的值。
- 重复到3,直到遍历完成,得到结果。
public static int findBottomLeftValue(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
int res = 0;
while (!queue.isEmpty()) {
int size = queue.size();
boolean isFirst = true;
while (size-- > 0) {
TreeNode node = queue.poll();
if (isFirst) {
res = node.val;
isFirst = false;
}
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
}
return res;
}
这道题可以稍微改变一下就是打印二叉树所有从左边看到的节点。
其实很简单,用一个List去保存每次得到的第一个节点:
public static List<Integer> findBottomLeftValue(TreeNode root) {
Queue<TreeNode> nodes = new LinkedList<TreeNode>();
nodes.add(root);
List<Integer> list = new ArrayList<>();
while(!nodes.isEmpty()) {
int size = nodes.size();
boolean isFirst = true;
while(size-- > 0 ) {
TreeNode node = nodes.poll();
if(isFirst) {
list.add(node.val);
isFirst = false;
}
if(node.left != null) {
nodes.add(node.left);
}
if(node.right != null) {
nodes.add(node.right);
}
}
}
return list;
}