糖逗 发表于 2021-4-1 14:47:57

C++刷leetcode(面试题 04.10. 检查子树)【树遍历】

题目描述:
检查子树。你有两棵非常大的二叉树:T1,有几万个节点;T2,有几万个节点。设计一个算法,判断 T2 是否为 T1 的子树。

如果 T1 有这么一个节点 n,其子树与 T2 一模一样,则 T2 为 T1 的子树,也就是说,从节点 n 处把树砍断,得到的树与 T2 完全相同。

注意:此题相对书上原题略有改动。

示例1:

输入:t1 = , t2 =
输出:true
示例2:

输入:t1 = , t2 =
输出:false
提示:

树的节点数目范围为。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/check-subtree-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


/**
* Definition for a binary tree node.
* struct TreeNode {
*   int val;
*   TreeNode *left;
*   TreeNode *right;
*   TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
private:
    TreeNode* store = NULL;
public:
    bool find_root(TreeNode* t1, TreeNode* t2){
      if(t1 == NULL)return false;
      if(t1 -> val == t2 -> val){
            store = t1;
            return true;
      }
      return find_root(t1 -> left, t2) || find_root(t1 -> right, t2);
    }
    bool euqal_tree(TreeNode* t1, TreeNode* t2){
      if(t1 == NULL && t2 == NULL)return true;
      else if(t1 == NULL && t2 != NULL)return false;
      else if(t1 != NULL && t2 != NULL){
            if(t1 -> val == t2 -> val){
                return euqal_tree(t1 -> left, t2 -> left) && euqal_tree(t1 -> right, t2 -> right);
            }
            else{
                return false;
            }
      }
      return false;
    }
    bool checkSubTree(TreeNode* t1, TreeNode* t2) {
      //树的遍历
      if(t1 == NULL && t2 != NULL)return false;
      else if(t1 == NULL && t2 == NULL)return true;
      else if(t1 != NULL &&t2 == NULL)return true;
      //先找到t1中的子节点
      if(find_root(t1, t2)){
            //比较两个树是否相等
            return euqal_tree(store, t2);
      }
      return false;
    }
};

糖逗 发表于 2021-4-1 22:19:32

{:10_323:}
页: [1]
查看完整版本: C++刷leetcode(面试题 04.10. 检查子树)【树遍历】