给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例 1:
输入: [1,2,3,null,5,null,4] 输出: [1,3,4] 示例 2:
输入: [1,null,3] 输出: [1,3] 示例 3:
输入: [] 输出: []
思考
LayerNodes:表示当前层节点数,初始化为1,代表第一层一个节点。 temp:用来记录下一层节点数
class Solution {
public List<Integer> rightSideView(TreeNode root) {
int LayerNodes = 1;
Queue<TreeNode> queue=new LinkedList<>();
List<Integer> list = new ArrayList<>();
if(root==null) return list;
queue.add(root);
while (!queue.isEmpty()){
int temp = 0;
while (LayerNodes!=0){
LayerNodes--;
TreeNode treeNode = queue.poll();
if(LayerNodes==0) list.add(treeNode.val);
if(treeNode.left!=null)
{
queue.add(treeNode.left);
temp++;
}
if(treeNode.right!=null)
{
queue.add(treeNode.right);
temp++;
}
}
LayerNodes=temp;
}
return list;
}
}
|