力扣 572 另一个树的子树（双递归）

#572 另一个树的子树

给定两个非空二叉树 s和 t，检验 s中是否包含和t具有相同结构和节点值的子树。s的一个子树包括 s的一个节点和这个节点的所有子孙。s也可以看做它自身的一棵子树。

示例 1:

给定的树 s:

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

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

示例 2:

给定的树 s：

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

   4
  / \
 1   2
返回 false。

思路：

• 判断一个树是否为另一个树的子树，显然需要用到递归。
• 判断是否是子树，就需要判断树是否相等，也是递归。
• 具体看代码注释。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
     
    public boolean isSametree(TreeNode s, TreeNode t){
     
        //两个树都为空，肯定相等。
        if(s == null && t == null) return true;
        //只有一个树为空，肯定不相等。
        if(s == null || t == null) return false;
        //如果数值不相等，肯定不相等。
        if(s.val != t.val) return false;
        //递归地比较左子树和右子树。需要左子树相等且右子树相等。
        return isSametree(s.left,t.left) && isSametree(s.right,t.right);
    }

    public boolean isSubtree(TreeNode s, TreeNode t) {
     
        //两个树都为空，肯定相等。
        if(s == null && t == null) return true;
        //递归到s空了，子树还没匹配到，就不是子树了。
        if(s == null && t != null) return false;
        //如果s等于t，那就是子树。
        if(isSametree(s,t)) return true;
        //递归比较左子树或者右子树。
        return isSubtree(s.left,t) || isSubtree(s.right,t);
    }
}