剑指 Offer 26. 树的子结构

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:

     3
    / \
   4   5
  / \
 1   2
给定的树 B：

   4 
  /
 1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false
示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true
限制：

0 <= 节点个数 <= 10000

 

 

解法：

首先缕清思路，第一步应先遍历A找到与B根节点有相同值的结点，第二步再判断此结点后续结点是否与B后续结点相同（即同步遍历A中与B根节点值相等的结点），若否回到第一步，直到B遍历完。

 其次想算法，既然一直在遍历树，除非有特别要求，平时自然用代码较简洁的递归方法。

class Solution {
public:
    

    bool isSubStructure(TreeNode* A, TreeNode* B) //找与B根结点值相同的结点
    {
       auto result = false;

       if(A != nullptr and B!= nullptr)　　　　　　//遍历完毕条件
       {
           if(Equal(A -> val,B->val))　　　　　　　//找到了，继续同步遍历比对后续结点
                result = DoesAHaveB(A,B);
            if(!result)　　　　　　　　　　　　　　　　
                result = isSubStructure(A->left,B);
            if(!result)
                result = isSubStructure(A->right,B);
       }
       return result;
    }

    bool Equal(int A,int B)
    {
        return (A == B?true:false);　　　　　　　　//此函数在本题无关紧要，因为此题树结点val为int，若类型为double，则由于精度问题改写这个函数就有必要了。
    }
        
    bool DoesAHaveB(TreeNode* A,TreeNode* B)　　//A与B根结点值相等，同步遍历之
    {
        if(B == nullptr)　　　　　　　　　　　　　　//B遍历完了
            return true;
        if(A == nullptr)　　　　　　　　　　　　　　//B未遍历完而A遍历完了
            return false;
        if(!Equal(A->val,B->val))　　　　　　　　//A、B均未遍历完但值不等
            return  false;
        
        return  DoesAHaveB(A->left,B->left)　　　//A、B指针指向的当前结点值相等，递归遍历后续结点。
                and 
                DoesAHaveB(A->right,B->right);
    }
};