问题描述:
给定一棵二叉树,返回所有重复的子树。对于同一类的重复子树,你只需要返回其中任意一棵的根结点即可。
两棵树重复是指它们具有相同的结构以及相同的结点值。
示例 1:
1
/ \
2 3
/ / \
4 2 4
/
4
下面是两个重复的子树:
2
/
4
和
4
因此,你需要以列表的形式返回上述重复子树的根结点。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-duplicate-subtrees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
执行结果:
代码描述:
思路:深度优先搜索,将树的分支构成字符串保存在ks中,并且以ks为键,存入map。每存入一次,同时查找map中是否存在该键,如果值为2,说明,已经有相同的了,然后把ks放入res中,记录。直到把所有的分支都递归找完,然后就退出结束。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
vector<TreeNode*> res;
map<string, int> m;
if(root == NULL) return res;
dfs(root, res, m);
return res;
}
string dfs(TreeNode* root, vector<TreeNode *> &res, map<string, int> &m)
{
if(root == NULL) return "";
string ss;
ss = root->val;
//此行整型和字符串直接相加竟然通过了。但是还是需要转换一下,以防其他编译器不能通过。
string ks = ss + "," + dfs(root->left, res, m) + "," + dfs(root->right, res, m);
m[ks]++;
if(m[ks] == 2)
res.push_back(root);
return ks;
}
};