题目:在一个二叉树中查找每一层中的最大值
示例:
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> largestValues(TreeNode root) {
List<Integer> list=new ArrayList<>();
Queue<TreeNode> queue=new LinkedList<>();
int length=0;
int max=0;
TreeNode node;
if(root==null)
return list;
else
{
list.add(root.val);
if(root.left!=null)
{
queue.add(root.left);
}
if(root.right!=null)
{
queue.add(root.right);
}
while(queue.size()>0)
{
length=queue.size();
for(int i=0;i<length;i++)
{
node=queue.poll();
if(i==0)
max=node.val;
else
max=max>node.val?max:node.val;
if(node.left!=null)
{
queue.add(node.left);
}
if(node.right!=null)
{
queue.add(node.right);
}
}
list.add(max);
}
}
return list;
}
}
算法思路:在进行深度遍历的时候,运用队列的话会方便很多,将每一层的节点都放入队列中,并且记录下每一层的节点的数量,这样就可以通过这个数来访问每一个节点。访问节点的时候同时将其子节点再次放入队列中,因此遍历的话只需要判断这个队列是否为空,就可以依次访问所有的层次的节点。