652. Find Duplicate Subtrees(寻找重复的子树)

本文介绍了一种使用哈希映射来查找二叉树中所有重复子树的方法,并通过递归序列化子树结构来实现。该方法只需遍历一次树即可完成任务,在效率上表现优秀。

题目描述

Given a binary tree, return all duplicate subtrees. For each kind of duplicate subtrees, you only need to return the root node of any one of them.
Two trees are duplicate if they have the same structure with same node values.
在这里插入图片描述

题目链接

https://leetcode.com/problems/find-duplicate-subtrees/

方法思路

class Solution {
    //Runtime: 13 ms, faster than 96.49%
    //Memory Usage: 61.5 MB, less than 30.00%
    Map<String, Integer> count;
    List<TreeNode> ans;
    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        count = new HashMap();
        ans = new ArrayList();
        collect(root);
        return ans;
    }
 //为什么要转换成字符串的形式再进行添加判定呢?,不是很懂。tricky!
    public String collect(TreeNode node) {
        if (node == null) return "#";
        String serial = node.val + "," + collect(node.left) + "," + collect(node.right);
        count.put(serial, count.getOrDefault(serial, 0) + 1);
        if (count.get(serial) == 2)
            ans.add(node);
        return serial;
    }
}
#include <iostream> #include <vector> #include <queue> #include <unordered_map> #include <string> using namespace std; struct HuffmanNode { int weight; char data; HuffmanNode* left; HuffmanNode* right; HuffmanNode(int w, char d = '\0') : weight(w), data(d), left(nullptr), right(nullptr) {} }; // 辅助函数:打印哈夫曼树(前序遍历) void printHuffmanTree(HuffmanNode* root, string prefix = "", bool isLeft = true) { if (root == nullptr) return; cout << prefix << (isLeft ? "├── " : "└── ") << root->data << ":" << root->weight << endl; printHuffmanTree(root->left, prefix + (isLeft ? "│ " : " "), true); printHuffmanTree(root->right, prefix + (isLeft ? "│ " : " "), false); } // 构建所有可能的哈夫曼树 vector<HuffmanNode*> buildAllHuffmanTrees(vector<int> weights) { vector<HuffmanNode*> result; if (weights.empty()) return result; // 如果只有一个节点,直接返回 if (weights.size() == 1) { result.push_back(new HuffmanNode(weights[0])); return result; } // 找到最小的两个权重 vector<int> min1Candidates, min2Candidates; int min1 = INT32_MAX, min2 = INT32_MAX; for (int w : weights) { if (w < min1) { min2 = min1; min2Candidates.clear(); min1 = w; min1Candidates.clear(); min1Candidates.push_back(w); } else if (w == min1) { min1Candidates.push_back(w); } else if (w < min2) { min2 = w; min2Candidates.clear(); min2Candidates.push_back(w); } else if (w == min2) { min2Candidates.push_back(w); } } // 枚举所有可能的最小权值对 for (int m1 : min1Candidates) { for (int m2 : min2Candidates) { vector<int> newWeights = weights; newWeights.erase(remove(newWeights.begin(), newWeights.end(), m1), newWeights.end()); newWeights.erase(remove(newWeights.begin(), newWeights.end(), m2), newWeights.end()); newWeights.push_back(m1 + m2); // 递归构建子树 vector<HuffmanNode*> subTrees = buildAllHuffmanTrees(newWeights); for (HuffmanNode* subTree : subTrees) { HuffmanNode* newNode = new HuffmanNode(m1 + m2); newNode->left = new HuffmanNode(m1); newNode->right = new HuffmanNode(m2); result.push_back(newNode); } } } return result; } int main() { vector<int> weights = {1, 2, 3, 4, 5}; // 示例权重数组 vector<HuffmanNode*> allTrees = buildAllHuffmanTrees(weights); cout << "All Possible Huffman Trees:" << endl; for (size_t i = 0; i < allTrees.size(); ++i) { cout << "Tree " << i + 1 << ":" << endl; printHuffmanTree(allTrees[i]); cout << endl; } // 清理内存 for (HuffmanNode* tree : allTrees) { // 简单递归删除树节点 if (tree) { if (tree->left) delete tree->left; if (tree->right) delete tree->right; delete tree; } } return 0; }修改代码使得编译过程不出错
最新发布
06-08
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值