199. 二叉樹的右視圖

給定一棵二叉樹,想象自己站在它的右側,按照從頂部到底部的順序,返回從右側所能看到的節點值。

示例:

輸入: [1,2,3,null,5,null,4]
輸出: [1, 3, 4]
解釋:

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

解題思路:層次遍歷樹中的每一個結點,同時result只記錄每一層中的一個結點,如果不是最右邊的結點更新即可.,
代碼實現


/**
 * 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> result;
        if(root == NULL)
        return result;
    queue<pair<TreeNode*,int> > Queue;
    Queue.push(make_pair(root,0));
    while(!Queue.empty())
    {
        TreeNode* node = Queue.front().first;
        int step = Queue.front().second;
        Queue.pop();
        if(step == result.size())
        result.push_back(node->val);
        else
        result[step] = node->val;
        if(node->left)
        {
            Queue.push(make_pair(node->left,step + 1));
        }
        if(node->right)
        {
            Queue.push(make_pair(node->right, step + 1));
        }
    }
    return result;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章