在每个树行中找最大值

 

题目：

您需要在二叉树的每一行中找到最大的值。

示例：

输入:

          1
         / \
        3   2
       / \   \  
      5   3   9

输出: [1, 3, 9]

分析：借助队列，将每一层的的节点以及孩子放在一个vector中，孩子永远放在当前节点的下一层，然后求出每一个vector的最大值

class Solution {
public:
    //用vector<vector<int>>保存每一行元素
    //用队列<Node*, int>保存当前节点和当前行数
    vector<int> largestValues(TreeNode* root) {
        queue<pair<TreeNode*, int>> q;
        vector<vector<int>> vv;
        vector<int> ret;
        
        TreeNode* cur=root;
        int i=0;
        int index=0;
        if(cur)
            q.push({cur, i});
        while(!q.empty())
        {
            pair<TreeNode*, int> front=q.front();//队首元素
            TreeNode* node=front.first;//队首元素的节点指针
            index=front.second;//队首元素的层数
            
            if(vv.size()<index+1)
            {
                vv.push_back({});
            }
            
            vv[index].push_back(node->val);//将对应层节点元素放在对应的vector中
                        q.pop();

            //i = index;
            //再将左右孩子带进去(注意带的时候，孩子的下标应该是index+1)
            if(node->left)
            {
                q.push({node->left, index+1});
            }
            if(node->right)
            {
                q.push({node->right, index+1});
            }
        }
        //将每一个vector中的最大值取出来放在ret中
        for(int i=0;i<vv.size();i++)
        {
            auto it=max_element(vv[i].begin(), vv[i].end());//拿到指向最大元素的迭代器
            ret.push_back(*it);//将迭代器指向的元素push到ret中
        }
        
        return ret;
    }