Leetcode 652:寻找重复子树

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

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

示例 1:

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

下面是两个重复的子树:

  2
 /
4

4

来源:力扣(LeetCode)题目链接:原题

这个题目可以让我们好好想想如何利用二叉树的前序、中序、后序遍历框架。关于各个遍历可以看:二叉树各个遍历递归非递归实现

这里我们要找出重复的子树,那么对于一个子树就要做两件事情:
1、知道自己长什么样
2、知道别的子树长什么样

对于第一件事情,我们要知道这个子树长什么样,那又要做两件事情:
1、知道左子树长什么样
2、知道右子树长什么样

当我们把左右两个子树都求出来了,再加上自己的值,就是这个子树的样子。在这里我们对树进行序列化方便记录子树的样子。

 string str = to_string(root->val) + "," + travel(root->left) + "," + travel(root->right);

这就是核心的代码,求出子树序列的样子。

然后就是查找是否重复,在这里我们可以利用unordered_map来存放我们的序列,并且每求到一个子树序列,就判断是否存在,若是,则加入结果中。

		if(mp[str]==1) res.push_back(root);//只有一个重复时加入结果
        mp[str]++; //计数+1

所以在此处它的递归遍历形式其实是和先序遍历很像的,先访问自己的值,再访问左右子树,但是改成类似中序和后序也很简单。

完整代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<TreeNode*> res;
    unordered_map<string, int> mp;
    vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
        travel(root);
        return res;
    }
    
    string travel(TreeNode* root){
        if(root==0) return "";
        //二叉树先序序列化
        string str = to_string(root->val) + "," + travel(root->left) + "," + travel(root->right);
        
        if(mp[str]==1) res.push_back(root);
        mp[str]++;

        return str;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值