ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

刷题-力扣-508. 出现次数最多的子树元素和

2021-08-29 18:01:50  阅读:208  来源: 互联网

标签:right nodeList nums 力扣 nowMax 508 TreeNode left 刷题


508. 出现次数最多的子树元素和

题目链接

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/most-frequent-subtree-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目描述

给你一个二叉树的根结点,请你找出出现次数最多的子树元素和。一个结点的「子树元素和」定义为以该结点为根的二叉树上所有结点的元素之和(包括结点本身)。

你需要返回出现次数最多的子树元素和。如果有多个元素出现的次数相同,返回所有出现次数最多的子树元素和(不限顺序)。

示例 1:
输入:

  5
 /  \
2   -3

返回 [2, -3, 4],所有的值均只出现一次,以任意顺序返回所有值。

示例 2:
输入:

  5
 /  \
2   -5

返回 [2],只有 2 出现两次,-5 只出现 1 次。

提示: 假设任意子树元素和均可以用 32 位有符号整数表示。

题目分析

  1. 根据题目描述获取子树和出现次数最多的和
  2. 求每棵子树的和存储在nums数组中
  3. 计算数组nums中出现次数最多的数字

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> findFrequentTreeSum(TreeNode* root) {
        vector<int> nums;
        if (!root) return nums;
        vector<int> res;
        int max = 0;
        int nowMax = 1;
        queue<TreeNode*> nodeList;
        nodeList.emplace(root);
        while (!nodeList.empty()) {
            if (nodeList.front()->left) nodeList.emplace(nodeList.front()->left);
            if (nodeList.front()->right) nodeList.emplace(nodeList.front()->right);
            nums.emplace_back(subTreeSum(nodeList.front()));
            nodeList.pop();
        }
        sort(nums.begin(), nums.end());
        for (int i = 1; i < nums.size(); ++i) {
            if (nums[i] == nums[i - 1]) {
                ++nowMax;
            } else {
                if (nowMax > max) {
                    res.clear();
                    res.emplace_back(nums[i - 1]);
                    max = nowMax;
                    nowMax = 1;
                } else if (nowMax == max) {
                    res.emplace_back(nums[i - 1]);
                    max = nowMax;
                    nowMax = 1;
                } else {
                    nowMax = 1;
                }
            }
        }
        if (nowMax > max) {
            res.clear();
            res.emplace_back(nums[nums.size() - 1]);
        } else if (nowMax == max) {
            res.emplace_back(nums[nums.size() - 1]);
        }
        return res;
    }

private:
    int subTreeSum(TreeNode* root) {
        if (!root) return 0;
        return root->val + subTreeSum(root->left) + subTreeSum(root->right);
    }

};

标签:right,nodeList,nums,力扣,nowMax,508,TreeNode,left,刷题
来源: https://www.cnblogs.com/HanYG/p/15203134.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有