652. 尋找重複的子樹
阿新 • • 發佈:2020-11-19
給定一棵二叉樹,返回所有重複的子樹。對於同一類的重複子樹,你只需要返回其中任意一棵的根結點即可。
兩棵樹重複是指它們具有相同的結構以及相同的結點值。
示例 1:
1
/ \
2 3
/ / \
4 2 4
/
4
下面是兩個重複的子樹:
2
/
4
和
4
因此,你需要以列表的形式返回上述重複子樹的根結點。
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/find-duplicate-subtrees
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
/** * 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: unordered_map<string,int>hash; vector<TreeNode*>res; string dfs(TreeNode* node){ if(!node)return "NULL"; string left=dfs(node->left); string right=dfs(node->right); string s=to_string(node->val)+','+left+','+right; if(++hash[s]==2)res.emplace_back(node); return s; } vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) { dfs(root); return res; } };