二叉树的右视图

2020-04-22  本文已影响0人  我知他风雨兼程途径日暮不赏

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-right-side-view

1. 题目

给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。

示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:

   1            <---
 /     \
2       3         <---
 \        \
  5        4       <---

2. JAVA解答

首先获取树深度(M),我们只需要将循环M次,每次获取每层的最右边节点,即后序遍历即可。


时间复杂度O(N*M)空间复杂度O(M)
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    /***
         * 获取树深度
         */
        int deepTree(TreeNode root){
            if(root==null){
                return 0;
            }
            if(root.left==null && root.right==null){
                return 1;
            }
            if(root.left!=null && root.right!=null){
                return Math.max(deepTree(root.left),deepTree(root.right))+1;
            }else if(root.left==null){
                return deepTree(root.right)+1;
            }else{
                return deepTree(root.left)+1;
            }
        }

        /***
         * 获取树的具体层的
         */
        public int treeGetLgorithm(TreeNode root,int layer){
            if(layer==0){
                return root.val;
            }
            int res = Integer.MIN_VALUE;
            if(root.right!=null){
                res = treeGetLgorithm(root.right,layer-1);
            }
            if(res==Integer.MIN_VALUE){
                if(root.left!=null){
                    return treeGetLgorithm(root.left,layer-1);
                }

            }
            return res;
        }
        public List<Integer> rightSideView(TreeNode root) {
            List<Integer> res = new ArrayList<>();
            if(root==null){
                return res;
            }
            res.add(root.val);
            int layer = deepTree(root);
            for(int i=1;i<layer;i++){
                res.add(treeGetLgorithm(root,i));
            }
            return res;
        }
}
上一篇 下一篇

猜你喜欢

热点阅读