【leetcode】199. 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].

思路：采用层次遍历，用两个队列存储结点，用一个队列存储一层的结点，先将最右端（即队列尾部结点）的值存入result中，然后依次遍历该队列中的结点，将这些结点的孩子结点放入另一个队列中。就这样遍历完所有的节点，result即为所求。

c++代码（3ms，37.32%）

/**
 * 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;
        queue<TreeNode*> q1,q2;
        if(root==NULL)
            return  result;
        q1.push(root);
        TreeNode* tmp;
        //进行层次遍历 
        while(!q1.empty() || !q2.empty()){
            //当两个队列都非空时
    		 if(!q1.empty()){
    		 	//当q1非空时，此时从q1中取结点，将孩子结点放入q2
    			 tmp=q1.back();  //取这一层最右边的结点
    			 result.push_back(tmp->val);
    			 while(!q1.empty()){
    			 	tmp=q1.front();
    			 	q1.pop();
    			 	if(tmp->left!=NULL)
    			 		q2.push(tmp->left);
    			 	if(tmp->right!=NULL)
    			 		q2.push(tmp->right);
    			 } 
    		 }else{
    		 	 tmp=q2.back();  
    			 result.push_back(tmp->val);
    			 while(!q2.empty()){
    			 	tmp=q2.front();
    			 	q2.pop();
    			 	if(tmp->left!=NULL)
    			 		q1.push(tmp->left);
    			 	if(tmp->right!=NULL)
    			 		q1.push(tmp->right);
    			 } //while
    		 }//else
    		 
        }
        return result;
    }
};

看了一下讨论组里面，大神的代码：

class Solution {
public:
    void recursion(TreeNode *root, int level, vector<int> &res)
    {
        if(root==NULL) return ;
        if(res.size()<level) res.push_back(root->val);
        recursion(root->right, level+1, res);
        recursion(root->left, level+1, res);
    }
    
    vector<int> rightSideView(TreeNode *root) {
        vector<int> res;
        recursion(root, 1, res);
        return res;
    }
};