Most Frequent Subtree Sum问题及解法

来源:互联网 发布:.name域名是什么 编辑:程序博客网 时间:2024/05/18 17:01

问题描述:

Given the root of a tree, you are asked to find the most frequent subtree sum. The subtree sum of a node is defined as the sum of all the node values formed by the subtree rooted at that node (including the node itself). So what is the most frequent subtree sum value? If there is a tie, return all the values with the highest frequency in any order.

示例:
Input:

  5 /  \2   -3
return [2, -3, 4], since all the values happen only once, return all of them in any order.

Input:

  5 /  \2   -5
return [2], since 2 happens twice, however -5 only occur once.

Note: You may assume the sum of values in any subtree is in the range of 32-bit signed integer.


问题分析:

将每棵树及子树的和都保存到hash表中,并记录出现次数最多的和,最后遍历hash表,即可找到答案。


过程详见代码:

/** * 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 maxi;vector<int> findFrequentTreeSum(TreeNode* root) {unordered_map<int, int> m;vector<int> res;        maxi = 0;bl(m, root);for (auto c : m){if (c.second == maxi)res.emplace_back(c.first);}return res;}int bl(unordered_map<int, int>& m, TreeNode* root){if (root == nullptr) return 0;int sum = bl(m, root->left) + bl(m, root->right) + root->val;if (!m.count(sum)) m[sum] = 1;else m[sum]++;maxi = max(maxi, m[sum]);return sum;}    };


原创粉丝点击