您的位置:首页 > 产品设计 > UI/UE

Tree——Find Largest Value in Each Tree Row

2017-03-12 10:49 337 查看

问题描述:

你需要在二叉树的每一行中找到最大值,并将所有的最大值放在vector中作为返回值。

解题思路:

从根节点开始,依次将每一层的节点从左到右放入vector,current指向当前访问的节点,last指向当前层的最后一个节点。然后把每一层的所有节点值放入一个vector,访问当前层结束后,找出最大的元素放入result中。

源代码如下:

class Solution {

public:

    vector<int> largestValues(TreeNode* root) {

        vector<int> result;

        if(root!=NULL)

        {

            vector<TreeNode* > vec;

            vec.push_back(root);

            int current=0;

            int last=1;

            while(current<vec.size())

            {

                last=vec.size();

                vector<int> vals;

                for(;current<last;current++)

                {

                    vals.push_back(vec[current]->val);

                    if(vec[current]->left!=NULL) vec.push_back(vec[current]->left);

                    if(vec[current]->right!=NULL) vec.push_back(vec[current]->right);

                }

                int max=vals[0];

                for(int i=1;i<vals.size();i++)

                {

                    if(vals[i]>max) max=vals[i];

                }

                result.push_back(max);

            }

        }

        return result;

    }

};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: