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

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

示例:

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

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

解法1。沿用BFS解决二叉树层序遍历的解法，q存储下一层的节点，p存储当前遍历的节点，每次将p的最后一个元素也即当前层最右边的元素放到res中，然后用队列的方式从左至右遍历当前层所有节点，并把节点的左右子节点放到q中，遍历完p就清空了，进入下一层的遍历，p重新被初始化，这里比较浪费空间，可以优化一下

class Solution(object):
    def rightSideView(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if not root:
            return []
        q = [root]
        res = []
        while q:
            p = q
            q = []
            if p:
                tmp = p[-1]
                res.append(tmp.val)
            while p:
                node = p.pop(0)
                if node:
                    if node.left:
                        q.append(node.left)
                    if node.right:
                        q.append(node.right)
        return res

解法2。只维护一个队列q，每轮只存储最后一个元素即当前层最右边的元素，q是个长度变化的队列，每次计算长度的时候只由当前实际长度决定，与后续增减无关，所以内层for loop将遍历当前层的节点并pop掉，同时存储下一层的节点

class Solution(object):
    def rightSideView(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if not root:
            return []
        q = [root]
        res = []
        while q:
            res.append(q[-1].val)
            n = len(q)
            for i in range(n):
                node = q.pop(0)
                if node.left:
                    q.append(node.left)
                if node.right:
                    q.append(node.right)
        return res