剑指 Offer 26. 树的子结构 思路与代码

 

 

 思路：

        由题意可知，我们需要判断树B是不是树A的子树。不管怎么样，肯定要在树A中找到与B树根节点val值相等的节点。可以选择递归或者非递归的遍历。找到节点值与树B根节点val值相等的节点。进入compare递归判断它们的结构是不是相等的（注意：在递归时如果某次递归B节点为空，那么不管A是不是空，肯定返回true。如果B节点不为空，判断如果A节点为空，那么肯定返回false。两个都不为空的情况下，判断两节点的值是不是相等的，如果是相等的，继续左节点、右节点调用compare递归判断。倘若不相等，两个子树结构肯定不一样。此时肯定返回false）。

       主方法中如果两个节点的值不相等，就递归进入A的左子节点和右子节点，继续判断。前序遍历。时间复杂度O（n），空间复杂度O（n）。

        有了思路可以很容易得到代码：

 

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(B==null || A==null){
            return false;
        }
        //利用递归
        //先使用前序遍历找节点与B树根节点相同的节点。
      if(A.val==B.val && (compare(A.left,B.left)&&compare(A.right,B.right))){
          return true;
      }
      return isSubStructure(A.left,B) || isSubStructure(A.right,B);
    }

   public boolean compare(TreeNode node,TreeNode B){
        if(B==null){
            return true;
        }
        if(node==null){
            return false;
        }
        if( node.val==B.val){
           return compare(node.left,B.left) && compare(node.right,B.right);
        }else{
            return false;
        }
   }
}