一、题目描述
给定一个二叉树的根节点,定义二叉树中的子树和为该子树中所有节点值的和。二叉树中每一个节点都可以为子树根节点,所得子树和也包括该节点,输出所给二叉树中出现频率最高的子树和值,若存在连个或者多个值出现频率相同且最高,则将其以任意顺序输出。
二、思路解析
二叉树中每一个节点都对应一棵子树,因此每一个节点也对应一个子树和的值。这题可分为两个过程解决,第一步是求出所给二叉树中每一个节点所对应的子树和的值,第二步是在这些子树和的值中找到出现频率最高的值。由于输出结果并不唯一,因此采用Vector的结构体存储子树和的值,最终输出vector。
第一步的思路主要是对二叉树进行后续遍历,由于每个节点的子树和值等于其左右孩子树和以及其本身相加得到,采用后续遍历从二叉树的叶节点开始计算,很容易得到每个节点对于的子树和,然后将计算出的每一个节点对于的子树和按照后续遍历的顺序存储在Vector中。接下来对Vector中的值进行排序,将出现频率最高的值依次push到另外一个Vector中,最后输出该Vector即可。思路行进起来比较清晰,只是需要注意一些边界情况,比如某子树为NULL或者输入的二叉树为NULL等情况。
三、代码实现
class Solution {
public:
vector<int> findFrequentTreeSum(TreeNode* root) {
treeAdd(root);
if(sum.size() == 1) {
out.push_back(root->val);
return out;
}
sort(sum.begin(),sum.end());
for(i = 0; i < sum.size(); i++) {
if(sum[i] == sum[i + 1]) {
count++;
} else {
if(temp == count) {
out.push_back(sum[i]);
} else if(temp < count) {
out.clear();
out.push_back(sum[i]);
}
temp = max(temp,count);
count = 1;
}
}
return out;
}
void treeAdd(TreeNode* root) {
if(!root) return;
treeAdd(root->left);
treeAdd(root->right);
if(root->left == NULL) {
addnum1 = 0;
} else {
addnum1 = root->left->val;
}
if(root->right == NULL) {
addnum2 = 0;
} else {
addnum2 = root->right->val;
}
root->val += addnum1 + addnum2;
sum.push_back(root->val);
}
private:
vector<int> out,sum;
int temp = 0,i,count = 1;
int addnum1,addnum2;
};