面试--leetcode-572-另一棵树的子树

题目

给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在，返回 true ；否则，返回 false 。

二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。

示例 1：

输入：root = [3,4,5,1,2], subRoot = [4,1,2]
输出：true

 大家一开始的想法肯定都是需要遍历root树，也需要遍历subRoot树，一个一个暴力搜索，才能去找到是否是子树关系。

在一开始我想的是，利用一个递归函数，去遍历root，然后如果找到值相等的节点，那么再去找遍历subRoot，但是这样是有问题的。没有办法保证两个树如果出现值相等的时候，会同步深入。并且如果这样的话，两个树之间的递归是递进的关系，我们需要的并列的关系。

思路

我们要去找的是root 树中和subRoot相同的树，所以我需要不断去遍历，或者说不断去切割 root 树，直到切割（找到）了完全一样的树。isSubtreeCheck 就是在找到了根节点值相同的情况下去继续遍历寻找。

我们要控制root树，和subRoot树的递归，保证是并列的。

主要分三种情况：

• subRoot 是 root左子树的子树
• subRoot 是 root右子树的子树
• subRoot 就是 root

我们在 isSubtree 函数之中控制 root 的三层递归 左子树 右子树 和其本身。

然后在isSubtreeCheck 函数之中 控制subRoot 的遍历。

我先具体讲一下 isSubtreeCheck 函数的作用，如果root 和 subRoot 的根节点值相同，也就是root.val == subRoot.val的话，那么我就要继续去遍历 root 树和 subRoot 树。

而 isSubtree 函数就是控制 root 的三层递归 左子树 右子树 和其本身。

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    //关键就是三种 true 的情况要搞懂 
    //subRoot 是 root左子树的子树 subRoot 是 root右子树的子树 subRoot 就是 root
    public boolean isSubtree(TreeNode root, TreeNode subRoot) {
        if(root == null) return false;
        if(isSubtreeCheck(root,subRoot)) return true;
        return isSubtree(root.left,subRoot)||isSubtree(root.right,subRoot);
    }
    public boolean isSubtreeCheck(TreeNode p, TreeNode q){
        if(p == null&&q == null){
            return true;
        }
        //要先写p和q如果为空的条件，要不然p.val != q.val这个地方会报空指针问题
        if( p == null||q == null||p.val != q.val){
            return false;
        }
        //检查如果根节点相同的情况下，左节点一不一样，右节点一不一样
        return isSubtreeCheck(p.left,q.left)&&isSubtreeCheck(p.right,q.right);
    }
}