【LeetCode-中等题】199. 二叉树的右视图

怎么还不写代码2023-09-03 20:04

文章目录

题目

方法一：层序遍历取每一层最后一个元素

java 复制代码

// 方法一 ：层序 + 集合(取每层子集合最后一个元素)
    // List<List<Integer>> Rlist = new ArrayList<>();
    // public List<Integer> rightSideView(TreeNode root) {
    //     if(root == null ) return new ArrayList<>();
    //     Queue<TreeNode> queue = new ArrayDeque<>();
    //     queue.offer(root); 
    //     while(!queue.isEmpty()){
    //         int count = queue.size();
    //         List<Integer> res = new ArrayList<>();
    //         for(int i=0;i<count;i++){
    //             TreeNode node = queue.poll();
    //             res.add(node.val);
    //         if(node.left != null) queue.offer(node.left);
    //         if(node.right != null) queue.offer(node.right);
    //         }
    //         Rlist.add(res);//每层节点集合加入到大集合
    //     }
    //       List<Integer> result = new ArrayList<>();//结果集
    //     for(List list : Rlist ){
    //         result.add((Integer)list.get(list.size()-1));//取每层集合最后一个元素
    //     }
    //     return result;
    // }
// 方法一（优化） ：在层序遍历的时候直接要每一层的最后一个元素加入集合  就不需要把一层所有节点的加入集合再去取每层集合最后一个了
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> Rlist = new ArrayList<>();//结果集
        if(root == null ) return Rlist;
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root); 
        while(!queue.isEmpty()){
            int count = queue.size();
            for(int i=0;i<count;i++){
                TreeNode node = queue.poll();
                if(i == count -1)//只取一层最后一个元素（count）
                Rlist.add(node.val);
            if(node.left != null) queue.offer(node.left);
            if(node.right != null) queue.offer(node.right);
            }
        }
        return Rlist;
    }

方法二：深度优先搜索

根右左的顺序取每层第一个被访问到的节点（depth == list.size()）

java 复制代码

   List<Integer> result = new ArrayList<>();
    public List<Integer> rightSideView(TreeNode root) {
       dfs(root,0);//起始递归 root  深度为0
       return result;
    }

    public void dfs(TreeNode tree ,int  depth){
        if(tree == null) return ;

        if(depth == result.size()) result.add(tree.val);//若当前深度 = 集合的大小，说明在该层还没加入任何元素，此时只需加入第一次访问的元素就行
        dfs(tree.right,depth+1);//切换到下一层继续找第一次访问的元素
        dfs(tree.left,depth+1);//切换到下一层继续找第一次访问的元素
    }