二叉树类题-Leetcode199题-二叉树的右视图

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

示例:

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-right-side-view
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

该题其实是求每一层树的最右结点。也就是通过层序遍历，记录每一层最后的一个数字，最后返回数组。
层序遍历通过定义队列，先进先出，然后将左右子树加入队列。
这道题要多定义一个遍历 层数，在该层数未添加数据时 添加进去，如果该层数有数据，则更新。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
    vector<int>res;//结果数组
    queue<pair<TreeNode*,int>>Q;//用Pair连接层数和结点
    if(root)
    {
        Q.push(make_pair(root,0));//首先加入root和第0层
    }
    while(!Q.empty())
    {
        TreeNode* tmp=Q.front().first;//获取队列第一个结点
        int depth=Q.front().second;//获取队列第一个结点的层数
        Q.pop();//弹栈
        if(res.size()==depth)//如果层数和结果数组一样大，说明是新层
        {
            res.push_back(tmp->val);
        }
        else//否则是老层里的新右结点
        {
            res[depth]=tmp->val;//更新
        }
        if(tmp->left)  //存在左子树 就加入队列
        {            
            Q.push(make_pair(tmp->left,depth+1));         
        }
        if(tmp->right)//同理
        {           
            Q.push(make_pair(tmp->right,depth+1));
        }
    }
    return res;
}
};