Python学习课堂笔记:寻找重复的子树

千锋Python讲堂 2019-12-11

python node string class list arraylist 存储 数组 ScreenShot

本期的Python学习课堂笔记:寻找重复的子树

题目:

给定一棵二叉树,返回所有重复的子树。对于同一类的重复子树,你只需要返回其中任意一棵的根结点即可。

两棵树重复是指它们具有相同的结构以及相同的结点值。

示例 1:

1
/ \
2 3
/ / \
4 2 4
/
4
下面是两个重复的子树:

2
/
4

4
因此,你需要以列表的形式返回上述重复子树的根结点。

screenshot

解题思路:

这就是一道考察二叉树遍历的题, 遍历的时候序列化作为 String 类型存入 Map, 若其为第二次出现即将该结点加入数组.

代码:
这里以后序遍历为例, 需要注意叶子结点应当规定一个特殊字符作为替代 null 结点, 这里用的是 ‘#’

Java:

class Solution {
public List findDuplicateSubtrees(TreeNode root) {
List list = new ArrayList<>();//待返回数组
if (root == null) return list;
Map map = new HashMap<>();//哈希映射查找重覆子结点
traverse(root, map, list, "");
return list;
}

//后序遍历

private String traverse(TreeNode node, Map map, List list, String tree) {
if (node == null) return tree + "#";
String left = traverse(node.left, map, list, tree);//递归左子孩子
String right = traverse(node.right, map, list, tree);//递归右子孩子
tree = tree + node.val + left + right;//每个子树路径

map.put(tree, map.getOrDefault(tree, 0) + 1);//存储每个子树路径
if (map.get(tree) == 2) list.add(node);//第二次出现相同路径时存储该结点
return tree;
}
}
Python:

class Solution:
def findDuplicateSubtrees(self, root: TreeNode) -> List[TreeNode]:
res = [] # 待返回数组
if not root: return res
hash_map = {} # 哈希映射查找重覆子结点
self.traverse(root, hash_map, res, '')
return res

# 后序遍历

def traverse(self, node: TreeNode, hash_map, res, tree):
if not node: return tree + '#'
left = self.traverse(node.left, hash_map, res, tree) # 递归左子孩子
right = self.traverse(node.right, hash_map, res, tree) # 递归右子孩子
tree = tree + str(node.val) + left + right # 每个子树路径
if tree in hash_map.keys(): # 存储每个子树路径
hash_map[tree] += 1
else:
hash_map[tree] = 1
if hash_map[tree] == 2: # 第二次出现相同路径时存储该结点
res.append(node)
return tree

有不清楚的地方可以留言,更多的Python学习课堂笔记也会继续给大家更新!

登录 后评论
下一篇
云栖号
8460人浏览
2020-03-04
相关推荐
0
0
0
2711