题目描述:
给你一个二叉树的根结点,请你找出出现次数最多的子树元素和。一个结点的「子树元素和」定义为以该结点为根的二叉树上所有结点的元素之和(包括结点本身)。
你需要返回出现次数最多的子树元素和。如果有多个元素出现的次数相同,返回所有出现次数最多的子树元素和(不限顺序)。
示例 1:
输入:
返回 [2, -3, 4],所有的值均只出现一次,以任意顺序返回所有值。
示例 2:
输入:
返回 [2],只有 2 出现两次,-5 只出现 1 次。
提示: 假设任意子树元素和均可以用 32 位有符号整数表示。
方法1:
主要思路:解题汇总链接
(1)使用后序遍历+哈希统计;
(2)为了计算各个子树的和,对于树从下到上进行遍历,组合出各个子树的和,避免重复计算;
(3)并在遍历的过程中,统计出各个和的出现频次;
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
//后序遍历
int back_order(TreeNode*root,unordered_map<int,int>&mp){
if(root==NULL){//终止条件
return 0;
}
//分别求出左右子树之和
int left_sum=back_order(root->left,mp);
int right_sum=back_order(root->right,mp);
++mp[left_sum+right_sum+root->val];//再加上当前结点的值,就是当前子树的和
return left_sum+right_sum+root->val;//返回当前子树的和
}
vector<int> findFrequentTreeSum(TreeNode* root) {
//处理特殊的情形
if(root==NULL){
return{};
}
if(root->left==NULL&&root->right==NULL){
return {root->val};
}
int max_freq=0;//统计最大频率
unordered_map<int,int> mp;//统计各个和的频率
back_order(root,mp);//后续遍历
for(auto&it:mp){//找出最大频率
max_freq=max_freq<it.second?it.second:max_freq;
}
vector<int>res;//存储结果
for(auto&it:mp){//存储最大频率对应的数字
if(it.second==max_freq){
res.push_back(it.first);
}
}
return res;
}
};