Python學習課堂筆記：尋找重復的子樹

本期的Python學習課堂筆記：尋找重復的子樹

題目:

給定一棵二叉樹，返回所有重復的子樹。對于同一類的重復子樹，你只需要返回其中任意 一棵的根結點即可。

兩棵樹重復是指它們具有相同的結構以及相同的結點值。

示例 1：

 1
 / \
 2 3
 / / \
 4 2 4
 /
 4

下面是兩個重復的子樹：

 2
 /
 4

和

 4

因此，你需要以列表的形式返回上述重復子樹的根結點。

解題思路：

這就是一道考察二叉樹遍歷的題, 遍歷的時候序列化作為 String 類型存入 Map, 若其為第二次出現即將該結點加入數組.

代碼:

這里以后序遍歷為例, 需要注意葉子結點應當規定一個特殊字符作為替代 null 結點, 這里用的是 ‘#’

Java:

class Solution {
 public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
 List<TreeNode> list = new ArrayList<>();//待返回數組
 if (root == null) return list;
 Map<String, Integer> map = new HashMap<>();//哈希映射查找重覆子結點
 traverse(root, map, list, "");
 return list;
 }
	//后序遍歷
 private String traverse(TreeNode node, Map<String, Integer> map, List<TreeNode> 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學習課堂筆記也會繼續給大家更新！