leetcode652 寻找重复的子树

labuladong

如何判断我们应该用前序还是中序还是后序遍历的框架？

还是那句话，根据题意，思考一个二叉树节点需要做什么，到底用什么遍历顺序就清楚了。

题目链接：

leetcode652 寻找重复的子树

本题思路：

如果你想知道以自己为根的子树是不是重复的，是否应该被加入结果列表中，你需要知道什么信息？

你需要知道以下两点：

1、以我为根的这棵二叉树（子树）长啥样？使用「后序遍历」框架，通过拼接字符串把二叉树序列化

2、以其他节点为根的子树都长啥样？借助HashMap，存每个节点的子树的序列化结果以及出现次数

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */

class Solution {

    // 记录所有子树以及出现的次数
    HashMap<String, Integer> subTreeNum = new HashMap<>();
    // 记录重复的子树根节点
    List<TreeNode> res = new LinkedList<>();

    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        traverse(root);
        return res;
    }

    // 辅助函数 描述二叉树
    public String traverse(TreeNode root) {
        //base case
        if(root == null) {
            return "#";
        }

        String left = traverse(root.left);
        String right = traverse(root.right);

        String subTree = left + ',' + right + ',' + root.val;

        // 若hashmap中有这个子树，则返回对应value值，否则返回0
        int number = subTreeNum.getOrDefault(subTree, 0);

        // 只有在重复时才记录子树根节点 且重复多次只能加入一次
        if(number == 1) {
            res.add(root);
        }

        // 子树存入hashmap 对应的出现次数加一
        subTreeNum.put(subTree, number + 1);
        return subTree;
    }
}