给定一个二叉树的 根节点
root
,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
树这类问题用的最多的就是递归,因为树具有天然的递归结构:
我们来分析一下题目,给定一棵树根结点,求出每层的最右边的结点,是不是我们可以换个角度想,是不是相当于求树每层的最右边的结点,将一整棵树分成一层一层的进行看待,这是不是与我们学过的树的层序遍历有关,来我们开始写代码:
因为我们既要关注结点的值,还要关注结点的所在的层数,一般碰到我们需要关注的该物体的属性在2个或者2个以上时,这时候我们就需要手动的去创建一个Piar对,对其进行封装
public class pair{
private TreeNode node;
private Integer level;
public pair(TreeNode node,Integer level){
this.level=level;
this.node=node;
}
}
树的层序遍历是通过BFS实现的,实现的数据结构肯定不能少
Queue queue=new LinkedList<>();
因为我们需要返回一个List
List> list=new ArrayList<>();
List path=new ArrayList<>();
然后就开始套用BFS模板进行解答
queue.add(start);
while(!queue.isEmpty()){
int val=queue.poll();
...
if(node.left!=null){
queue.add(...);
}
if(node.right!=null){
queue.add(...);
}
通过BFS遍历之后,我们就会得到一个类似于这样的存储结点的链表,是不是我们所需要的是每个链表中的最后一个元素,我们将每个小链表拿出来,将每个中最后一个元素取出放到一个新的链表中,那么这个新的链表不就是我们想要求的最终列表链表么?
List result=new ArrayList<>(list.size());
for (List item:list) {
result.add(item.get(item.size()-1));
}
第一种方法的源码:
public List rightSideView(TreeNode root) {
List> list=new ArrayList<>();
List path=new ArrayList<>();
if(root==null){
return path;
}
Queue queue=new LinkedList<>();
queue.add(new pair(root,0));
while(!queue.isEmpty()){
pair pair=queue.poll();
TreeNode node=pair.node;
Integer level=pair.level;
if(list.size()==level){
list.add(new ArrayList<>());
}
List list2=list.get(level);
list2.add(node.val);
if(node.left!=null){
queue.add(new pair(node.left,level+1));
}
if(node.right!=null){
queue.add(new pair(node.right,level+1));
}
}
List result=new ArrayList<>(list.size());
for (List item:list) {
result.add(item.get(item.size()-1));
}
return result;
}
public class pair{
private TreeNode node;
private Integer level;
public pair(TreeNode node,Integer level){
this.level=level;
this.node=node;
}
}
这样一来我们这道题的解决完了,下面给大家提供一种新的思路(递归)
因为树是天然的递归结构,每层的最右边的结点是不是一般在某棵数的右子树(存在的时候)中,所以我们可以先递归右子树、后递归左子树
DG(root.right);
DG(root.left);
我们怎么才能获得最右边的节点呢?这样是不是和我们刚才讲的思路相差不多,树的高度,我们可以在递归的时候将树的高度当做参数传下去,然后我们就可以写成:
private void DG(TreeNode root, int depth) {
DG(root.right,depth+1);
DG(root.left,depth+1);
}
递归的两要素:递归结束条件+递归操作
那么是不是当我们这个结点为空时,直接return出去就好了
if(root==null){
return;
}
我们还应该声明一个容器去存储我们得到的结果,直接看源代码:
List list=new ArrayList<>();
public List rightSideView(TreeNode root) {
if(root==null){
return list;
}
DG(root,0);
return list;
}
private void DG(TreeNode root, int depth) {
if(root==null){
return;
}
if(depth==list.size()){
list.add(root.val);
}
DG(root.right,depth+1);
DG(root.left,depth+1);
}
这里一定有人回问?这里为什么是当depth==list.size()的时候,会将这个结点加入list中去,这种问题对于读递归不是很熟的东西确实是比较难以捉摸的,原因是这样的,因为我们在递归的时候是优先递归的是右子树,后递归的左子树,所以每次都是每层的最右边的那个节点优先到达的这层,所以才往我们的结果集中加入当前节点,所以才会这么写