LeetCode题解（Week5）: 515. Find Largest Value in Each Tree Row

* 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:
    void searchTree(TreeNode* curr,int level, vector<int>& res){
        // 判断目前vector中当前层的元素是不是最大值
        if(res.size() > level && res[level] < curr->val)
            res[level] = curr->val;
        // 由于level是递增的，不会发生跳变，因此可以这样做
        if(res.size() <= level)
            res.push_back(curr->val);
        //递归
        if(curr->left)
            searchTree(curr->left,level+1,res);
        if(curr->right)
            searchTree(curr->right,level+1,res);
    }

    vector<int> largestValues(TreeNode* root) {
        vector<int> res;

        if(root)//考虑到输入有可能是一个空树
            searchTree(root,0,res);//递归根节点

        return res;
    }
};

解析

这到题目的核心在于找出每一个节点所在的层。如果能把对应的层数找出来，求最大值也就好办。在这里我用了DFS，在递归的时候通过一个函数参数来记录当前所在的层数。每次递归时，找到当前节点的值以及对应层数。如果层数对应的元素还没有在vector则通过push_back添加，如果vector已经有对应层数，就比较该元素值与当前的节点值。若左右子节点不为空，则进行递归。

分析算法复杂度：这道题目不需要额外的空间去存储变量，若有N个树节点，就需要递归N次，因此算法复杂度为O(N)