算法练习

二叉树的右视图(LeetCode 199 --- 广度优先搜索)

2020-02-24  本文已影响0人  倚剑赏雪

题目

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

示例:

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

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

解析

思考与分析

这个和二叉树的层序遍历区别在于,只需要保存最右边的TreeNode即可

代码

 public IList<int> RightSideView(TreeNode root) {
        IList<int>  view =new List<int>();//按层搜索的最后一个节点
        if(root==null)
            return view;
        int level=0;
        Queue<TreeNode> queue = new Queue<TreeNode>();
        queue.Enqueue(root);
        while ( queue.Count > 0 )
        {
            int levelLength = queue.Count;
            for(int i = 0; i < levelLength; ++i) {
                if(i==levelLength-1){
                    view.Add(queue.Peek().val);
                }
                TreeNode node = queue.Dequeue();
            
                if (node.left != null) queue.Enqueue(node.left);
                if (node.right != null) queue.Enqueue(node.right);
            
            }        
            level++;
        }
        return view;
    }
上一篇 下一篇

猜你喜欢

热点阅读