9.9 力扣652. 寻找重复的子树

最新推荐文章于 2022-11-12 10:38:41 发布

en7oy

最新推荐文章于 2022-11-12 10:38:41 发布

阅读量812

点赞数

分类专栏：每日一题文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/harne33/article/details/126778557

版权

每日一题专栏收录该内容

51 篇文章 0 订阅

订阅专栏

652. 寻找重复的子树 - 力扣（LeetCode）

给定一棵二叉树 root，返回所有重复的子树。

对于同一类的重复子树，你只需要返回其中任意一棵的根结点即可。

如果两棵树具有相同的结构和相同的结点值，则它们是重复的。

树中的结点数在[1,10^4]范围内。
-200 <= Node.val <= 200

示例 1：

输入：root = [1,2,3,4,null,2,4,null,null,4]
输出：[[2,4],[4]]

示例 2：

输入：root = [2,2,2,3,null,3,null]
输出：[[2,3],[3]]

思路：首先肯定要遍历一遍树，要有一个方法去判断两个子树是否一样，而判断子树是否一样的前提条件是根节点的val值一样或者用哈希表去保存所有的子树（利用序列号成string），子树每次插入哈希表前判断表中是否存在与该子树相同的string，有则是重复子树

/**
 * 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:
    string dfs(TreeNode* root)
    {   
        if(root==nullptr) return " ";
        //记录完整子树（子树序列号）
        string temp=to_string(root->val)+"("+dfs(root->left)+")"+"("+dfs(root->right)+")";
        auto it=m.find(temp);
        if(it!=m.end())
        {
            //插入根节点（要考虑后面还有相同子树时插入的会是相同节点，为了避免ret出现相同子树）
            ret.insert(it->second);
        }
        else
        {
            m[temp]=root;
        }
        return temp;
    }
    vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
        dfs(root);
        return vector<TreeNode*>(ret.begin(),ret.end());
    }
private:
    unordered_map<string,TreeNode*> m;
    //返回的也要去重
    unordered_set<TreeNode*> ret;
};