ITPub博客

首页 > > > [LeetCode] Binary Tree Right Side View

[LeetCode] Binary Tree Right Side View

作者:Ruiyuan Li 时间:2015-04-06 00:00:00 0 删除 编辑

Binary Tree Right Side View

Given a binary tree, imagine yourself standing on the right side of it, return the values of the nodes you can see ordered from top to bottom.

For example:
Given the following binary tree,

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


You should return [1, 3, 4].

解题思路:

这道题较为简单,可以用类似于先序遍历的办法来做,但是有点不同,遍历的顺序是根,右,左。用一个map记录每一层第一个出现的值。下面是代码:

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector rightSideView(TreeNode *root) {
        map levelToVal;   //记录每一层第一个出现的数值。
        tranver(root, 0, levelToVal);
        
        vector result;
        
        for(map::iterator it=levelToVal.begin(); it!=levelToVal.end(); it++){
            result.push_back(it->second);
        }
        
        return result;
    }
    
    //类似于先序遍历,但是,顺序是上,右,左
    void tranver(TreeNode* node, int level, map& levelToVal){
        if(node==NULL){
            return;
        }
        map::iterator it=levelToVal.find(level);
        if(it==levelToVal.end()){
            levelToVal.insert(map::value_type(level, node->val));
        }
        tranver(node->right, level+1, levelToVal);
        tranver(node->left, level+1, levelToVal);
    }
};

二次刷题2015-10-10

其实可以直接按照线序遍历来做。因为后面的值会覆盖前面的值。

/**
 * 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 rightSideView(TreeNode* root) {
        vector result;
        map m;
        helper(root, m, 0);
        for(map::iterator it = m.begin(); it!=m.end(); it++){
            result.push_back(it->second);
        }
        return result;
    }
    
    void helper(TreeNode* root, map& m, int level){
        if(root == NULL){
            return;
        }
        m[level] = root->val;
        helper(root->left, m, level + 1);
        helper(root->right, m, level + 1);
    }
};


<!-- -->

原文作者:;如有侵权,请联系管理员删除。

请登录后发表评论 登录
全部评论

注册时间:2020-05-21

  • 博文量
    495
  • 访问量
    4826