1. 程式人生 > 實用技巧 >652. 尋找重複的子樹

652. 尋找重複的子樹

給定一棵二叉樹,返回所有重複的子樹。對於同一類的重複子樹,你只需要返回其中任意一棵的根結點即可。

兩棵樹重複是指它們具有相同的結構以及相同的結點值。

示例 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; } };