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

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);
    }

};