JAVA 牛牛

508. 出现次数最多的子树元素和

  • May 08, 2020
  • 0 views

题目 508. 出现次数最多的子树元素和

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

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

示例 1：

输入:

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

示例 2：

输入：

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

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

方法一 递归+哈希表

解题思路

代码

java

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    // 配合map 记录最多出现次数的子树
    private int max = 0;

    public int[] findFrequentTreeSum(TreeNode root) {
        if (root == null)
            return new int[0];
        Map<Integer, Integer> map = new HashMap<>();
        helper(root, map);
        // 然后求出map中value最大值对应的Key
        List<Integer> res = new LinkedList<>();
        for (Integer i : map.keySet()) {
            if (map.get(i) == max)
                res.add(i);
        }
        // 转数组
        int[] resArr = new int[res.size()];
        for (int i = 0; i < res.size(); i++) {
            resArr[i] = res.get(i);
        }
        return resArr;
    }

    public int helper(TreeNode root, Map<Integer, Integer> map) {
        if (root == null)
            return 0;
        int left = helper(root.left, map);
        int right = helper(root.right, map);
        // 当前节点为根的元素和
        int val = left + right + root.val;
        map.put(val, map.getOrDefault(val, 0) + 1);
        max = Math.max(max, map.get(val));
        return val;
    }
}

python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def findFrequentTreeSum(self, root: TreeNode) -> List[int]:

        # 生成默认的字典 相当于 java中的 Map.getOrDefault(key,0)
        tree_dict = collections.defaultdict(int)

        # 递归 遍历出所有 子的元素和 并用 字典记录
        def helper(root):
            if not root:return 0
            left = helper(root.left)
            right = helper(root.right)
            temp = left + right + root.val
            tree_dict[temp] += 1
            return temp
        if not root:return []
        helper(root)

        # 记录出现最多次的 子树的元素和
        tree_max = 0
        # python 通过 dict.values() 遍历出字典中的 value
        # 类似于 java中的 for(Map.Entry<String, String> entry : map.entrySet())
        for cnt in tree_dict.values():
            tree_max = max(cnt,tree_max)
        res_list = []

        # python 通过 dict.items() 遍历出字典中的key 和 value
        for key, val in tree_dict.items():
            if val == tree_max:
                res_list.append(key)
        return res_list
Search