LeetCode 508. 出现次数最多的子树元素和(递归)

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

发布了914 篇原创文章 · 获赞 2966 · 访问量 50万+

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/105715277
今日推荐