题目
给你两棵二叉树 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);
}
}