leetcode652. 寻找重复的子树树的序列化

最新推荐文章于 2024-09-09 15:49:20 发布

原创最新推荐文章于 2024-09-09 15:49:20 发布 · 264 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#深度优先 #算法

笔记专栏收录该内容

122 篇文章

订阅专栏

给定一棵二叉树，通过先序遍历序列化每个子树并利用哈希表记录出现次数，找到重复的子树。当哈希表中的计数为2时，将对应的节点添加到结果集中。

https://leetcode.cn/problems/find-duplicate-subtrees/?
给你一棵二叉树的根节点 root ，返回所有重复的子树。
对于同一类的重复子树，你只需要返回其中任意一棵的根结点即可。
如果两棵树具有相同的结构和相同的结点值，则认为二者是重复的。

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

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


输入：root = [2,1,1]
输出：[[1]]


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

题解

遍历树的每个节点，将节点子树按照先序或后续遍历序列化成字符串，作为哈希表的键。哈希值为出现次数。

code

/**
 * 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 {
private:
    unordered_map<string,int> mymap;
    vector<TreeNode*> ans;
public:
    string dfs(TreeNode* node){
        if(node==nullptr) return "";
        //先序化成字符串，注：中序遍历，【2,2,null】和【2,null,2】的返回是相同的，所以无法用于序列化
        string key=to_string(node->val)+" "+dfs(node->left)+" "+dfs(node->right);
        mymap[key]++;
        if(mymap[key]==2) ans.push_back(node);
        return key;
    }
    vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
        dfs(root);//遍历树的每一个节点
        return ans;
    }
};