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]

---

思路

找到二叉树的每一行中的最大值，然后返回一个vector。
可以很容易想到用层序遍历来实现，遍历每一层保存每一层的最大值。需要用一个值来保存当前层的节点个数，然后将子节点放入一个队列中，从而实现每一层的遍历。
是用一个容器vec来做辅助队列，当vec大小为整个树所有节点数时，说明遍历结束。用一个数cur来辅助这一过程，每遍历一个节点cur加一，并将该节点的左右子节点压入vec中，遍历结束时，树中全部节点都在vec中，而cur为全部节点数。
在遍历每一层时，用i来表示每一层的节点数，用max表示这一层的最大值，访问每个节点时更新max，这一层访问结束后，将max存入结果容器res中。
需要注意的时两层循环之间的关系，以及各自的结束条件。

---

代码

/**
 * 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> res;
        if(root == NULL)
            return res;
        vector<TreeNode*> vec; //辅助队列用于遍历全部节点

        vec.push_back(root);

        int cur = 0;
        int i = 0; 

        while(cur<vec.size())
        {
            i = vec.size(); //当前层的节点个数
            int max = vec[cur]->val; //用来保存当前层最大值
            while(cur<i)
            {
                if(max < vec[cur]->val)
                    max = vec[cur]->val;
                if(vec[cur]->left != NULL)
                    vec.push_back(vec[cur]->left);
                if(vec[cur]->right != NULL)
                    vec.push_back(vec[cur]->right);
                cur++;
            }
            res.push_back(max);
        }
        return res;
    }
};