给定一棵二叉树,返回所有重复的子树。对于同一类的重复子树,你只需要返回其中任意一棵的根结点即可。
两棵树重复是指它们具有相同的结构以及相同的结点值。
思路
首先是进行序列化,然后序列化的过程中,进行树的深度优先搜索,在树的搜索过程中,进行一个结果的统计。
毕竟树的深度优先访问的时间负杂度是O(n)的。
代码
class Solution {
public:
vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
if (root == nullptr) return {
};
vector<TreeNode*> res;
map<string, int> m;
dfs(root, res, m);
return res;
}
string dfs(TreeNode* root, vector<TreeNode*>&res, map<string,int>&m) {
if (root == nullptr) return "";
string ks = to_string(root->val) + "," + dfs(root->left,res,m) + "," + dfs(root->right, res, m);
if (++m[ks] == 2) {
res.push_back(root);
}
return ks;
}
};
总结
树的遍历过程+操作就可以完成很多任务了。