Leetcode 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> ret;  // 存放结果值
        if(root == nullptr)
        {
            return ret;    // 如果根节点指向空指针，则返回空。
        }
        dfs(root,0,ret);   //调用深度优先搜索函数,先序遍历搜索
        return ret;
    }
private:  
    void dfs(const TreeNode* root, const int depth, vector<int> &res)// 深度优先搜索
    {
        if (root == nullptr)
           return;                //如果是叶节点则返回
        if (depth == res.size())  //如果此时的树深等于ret的长度，则将节点的值添加进去
        {
             res.push_back(root->val);
        }
        else//如果此时的深度不等于res的长度，说明这一层已经写进入过值，所以要进行比较
        {
            res[depth] = max(res[depth], root->val);
        }
        dfs(root->left, depth + 1, res);
        dfs(root-> right, depth + 1, res);
    }
};

 有几点需要注意的：

1. 深度优先搜索时的参数，depth是实参调用，每次函数里边的值是不改变上一层中的值，而根节点指针和结果res都是会变化的，res的长度是 引用传参。

2. 此时为先序遍历