LeetCode·每日一题·652.寻找重复的子树·递归·序列化

题目

 

示例

 

思路

在做本题之前，建议先看

[二叉树的序列化和反序列化](https://leetcode.cn/problems/xu-lie-hua-er-cha-shu-lcof/solution/by-xun-ge-v-db0n/)

[哈希算法-C语言具体实现详解](http://t.csdn.cn/0UoNg)

看了上面两个相信这题也就不是上面问题了

**具体实现**

将树按节点对其进行序列化，然后加入hash中，如果之前存在过相同的序列化，那么肯定是重复的子树，**注意**这里如果多次出现只保存一次，最后返回即可

代码注释超级详细

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

struct my_struct {//散列表
    char name[50000];          /* key */
    int id;//相同key出现的次数，可以自己定义意义
    UT_hash_handle hh;         /* makes this structure hashable */
};

struct my_struct *users = NULL;//申请哈希表

//将树节点之后的子树节点序列化
//每个人的都可以不一样
void dfs(struct TreeNode * root, char * str)
{
    if(root == NULL)//空节点，用@代替
    {
        strcat(str, "@");
        strcat(str, ",");
        return;
    }
    char s[7] = "";//临时保存树元素值
    sprintf(s, "%d,", root->val);//类型转换
    strcat(str, s);//加入字符串数组中
    dfs(root->left, str);//递归遍历左右子节点
    dfs(root->right, str);
    return;
}

//将所有节点序列化，比加入哈希表
void ifTree(struct TreeNode ** ans, int * returnSize, struct TreeNode * root)
{
    if(!root) return;
    char str[50000] = {0};
    dfs(root, str);//序列化
    struct my_struct *s = NULL;
    HASH_FIND_STR(users, str, s);//判断序列化字符串是否在hash中存在
    if(s)//存在
    {
        if(s->id == 1)//判断是否之前只出现过一次，如果出现过多次
                      //之前肯定已经保存过了，不需要在保存了
        {
            ans[(*returnSize)++] = root;
        }
        s->id++;//出现次数+1
    }
    else//没有出现过
    {
        s = (struct my_struct *)malloc(sizeof *s);//加入哈希表
        strcpy(s->name, str);
        s->id = 1;
        HASH_ADD_STR(users, name, s);
    }
    ifTree(ans, returnSize, root->left);//序列化左右子树
    ifTree(ans, returnSize, root->right);
    return;
}

struct TreeNode** findDuplicateSubtrees(struct TreeNode* root, int* returnSize){
    struct TreeNode ** ans = (struct TreeNode **)malloc(sizeof(struct TreeNode *) * 10000);
    *returnSize = 0;
    ifTree(ans, returnSize, root);//初始化变量

    /* free the hash table contents */
    struct my_struct *tmp = NULL;//销毁哈希表
    struct my_struct *s = NULL;
    HASH_ITER(hh, users, s, tmp) {
      HASH_DEL(users, s);
      free(s);
    }
    return ans;
}




作者：xun-ge-v
链接：https://leetcode.cn/problems/find-duplicate-subtrees/solution/-by-xun-ge-v-w3jj/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。