515. Find Largest Value in Each Tree Row

题目

You need to find the largest value in each row of a binary tree.

Example:

Input: 

          1
         / \
        3   2
       / \   \  
      5   3   9 

Output: [1, 3, 9]

分析

递归遍历树，遍历过程中将结果向量大小与当前深度比较，小于则说明当前深度第一次遍历到，压入结果向量，否则与结果向量中的值比较，保留最大的。

/**
 * 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> largestValues(TreeNode* root) {
        vector<int> rs;
        if(root==NULL)
            return rs;
        rs.push_back(root->val);
        deepsearch(root->left,rs,0+1);
        deepsearch(root->right,rs,0+1);
        return rs;
    }
    void deepsearch(TreeNode* root,vector<int>& rs,int depth)
    {
        if(root==NULL)
            return;
        if(rs.size()<depth+1)//如果rs小于depth+1，说明当前深度的结果未保存
            rs.push_back(root->val);
        else
            rs[depth]=max(rs[depth],root->val);//否则说明当前深度的结果已存在，与当前值比较选择最大的保存
        deepsearch(root->left,rs,depth+1);
        deepsearch(root->right,rs,depth+1);
    }
};