1. 题目
给你一个二叉树的根结点,请你找出出现次数最多的子树元素和。
一个结点的「子树元素和」定义为以该结点为根的二叉树上所有结点的元素之和(包括结点本身)。
你需要返回出现次数最多的子树元素和。如果有多个元素出现的次数相同,返回所有出现次数最多的子树元素和(不限顺序)。
示例 1:
输入:5/ \
2 -3
返回 [2, -3, 4],所有的值均只出现一次,以任意顺序返回所有值。示例 2:
输入:5/ \
2 -5
返回 [2],只有 2 出现两次,-5 只出现 1 次。提示: 假设任意子树元素和均可以用 32 位有符号整数表示。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/most-frequent-subtree-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
- 自底向上,计算每个节点的子树和,存入哈希map,同时记录最大出现次数
- 遍历哈希map取出最大次数的值
class Solution {unordered_map<int,int> m;vector<int> ans;int maxCount = 0;int s;
public:vector<int> findFrequentTreeSum(TreeNode* root) {sumof(root);for(auto& mi : m)if(mi.second == maxCount)ans.push_back(mi.first);return ans;}int sumof(TreeNode* root){if(root == NULL)return 0;int l = sumof(root->left);//自底向上int r = sumof(root->right);s = l+r+root->val;if(++m[s] > maxCount)maxCount = m[s];return s;}
};
32 ms 23 MB