Description
Given a binary tree, return all duplicate subtrees. For each kind of duplicate subtrees, you only need to return the root node of any one of them.
Two trees are duplicate if they have the same structure with same node values.
Example 1:
1
/ \
2 3
/ / \
4 2 4
/
4
The following are two duplicate subtrees:
2
/
4
and
4
Therefore, you need to return above trees’ root in the form of a list.
问题描述
给定二叉树, 返回所有重复子树。对于每类重复子树, 只需要返回任意一个子树的根节点。
两棵树如何结构相同且所有对应节点的值相等, 那么这两棵树重复。
问题分析
后序遍历, 通过字符串表示一棵树, 通过map保存字符串对应个数, 若个数大于1, 说明此树重复, 可以添加入结果中
解法
class Solution {
public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
HashMap<String, Integer> map = new HashMap();
List<TreeNode> res = new ArrayList();
helper(root, map, res);
return res;
}
private String helper(TreeNode root, HashMap<String, Integer> map, List<TreeNode> res){
if(root == null) return "#";
String temp = root.val + helper(root.left, map, res) + helper(root.right, map, res);
if(!map.containsKey(temp)){
map.put(temp,1);
}else{
if(map.get(temp) < 2){
map.put(temp, 2);
res.add(root);
}
}
return temp;
}
}