2021-03-15 | 508. 出现次数最多的子树元素和

ゞ浴缸里的玫瑰 2022-11-12 01:45 73阅读 0赞

### 1. 题目描述 ###

给你一个二叉树的根结点，请你找出出现次数最多的子树元素和。一个结点的「子树元素和」定义为以该结点为根的二叉树上所有结点的元素之和（包括结点本身）。

你需要返回出现次数最多的子树元素和。如果有多个元素出现的次数相同，返回所有出现次数最多的子树元素和（不限顺序）。

**示例 1：**

输入:
      5
     /  \
    2   -3
    返回 [2, -3, 4]，所有的值均只出现一次，以任意顺序返回所有值。

**示例 2：**

输入：
      5
     /  \
    2   -5
    返回 [2]，只有 2 出现两次，-5 只出现 1 次。

**提示：** 假设任意子树元素和均可以用 32 位有符号整数表示。

### 2. 解题思路 ###

这道题和二叉树的众数那道题是一样的思路：

*  首先，先遍历出一次树， 求所有节点的子树和
 *  在遍历的过程中，使用map来记录每个元素出现的次数
 *  遍历完成之后，遍历map，找出次数最多的和，放在res中即可

**复杂度分析：**

*  时间复杂度：O(n)，其中n是这棵树的节点数量，我们需要遍历整棵树，来求每个节点的子树和。
 *  空间复杂度：O(n)，其中n是这棵树的节点数量，这里需要的是递归的栈空间的空间代价。

### 3. 代码实现 ###

/** * Definition for a binary tree node. * function TreeNode(val) { * this.val = val; * this.left = this.right = null; * } */
    /** * @param {TreeNode} root * @return {number[]} */
    var findFrequentTreeSum = function(root) { 
        let map = { }, res = [], max = 0
    
        const calcuSum = (root) => { 
            if(!root){ 
                return 0
            }
            let left = calcuSum(root.left)
            let right = calcuSum(root.right)
    
            let sum = left + right + root.val
            // 将当前节点赋值为其所有子节点的和，方便后面进行计算
            root.val = sum
            map[sum] ? map[sum] += 1 : map[sum] = 1
            return root.val
        }
        calcuSum(root)
    
        for(let key in map){ 
            if(map[key] === max){ 
                res.push(key)
            }
            if(map[key] > max){ 
                max = map[key]
                res = [key]
            }
        }
        return res
    };

### 4. 提交结果 ###

![在这里插入图片描述][20210321210622858.png]

[20210321210622858.png]: /images/20221022/f1e68c0742834446943a77ce00ae233c.png?