leetcode刷题（6）—— 另一个树的子树

题目

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

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

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

输入：root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2]
输出：false

提示：
root 树上的节点数量范围是 [1, 2000]
subRoot 树上的节点数量范围是 [1, 1000]
-104 <= root.val <= 104
-104 <= subRoot.val <= 104

解法

采用递归的思路，先从根节点开始判断，如果有其中一个为null都应该返回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 isSubtree(TreeNode s, TreeNode t) {
       boolean result = false;
        if(s == null || t == null) return false;
        result = isNodeEquals(s, t);
    
        
        if (!result){
            result = isSubtree(s.left,t);
        }
        
        if (!result){
            result = isSubtree(s.right,t);
        }
        
        return result;
    }
    
     public boolean isNodeEquals(TreeNode s, TreeNode t) {
     if (t == null) {
            return s == null;
        }
        if (s == null) {
            return t == null;
        }
        if (s.val != t.val) {
            return false;
        }
        return isNodeEquals(s.left, t.left) && isNodeEquals(s.right, t.right);
    }
}

// 定义二叉树节点类
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class SubtreeOfAnotherTree {
    // 判断 t 是否是 s 的子树
    public boolean isSubtree(TreeNode s, TreeNode t) {
        if (s == null) {
            return false;
        }
        // 检查当前节点 s 是否和 t 是相同的树
        if (isSameTree(s, t)) {
            return true;
        }
        // 递归检查 s 的左子树和右子树
        return isSubtree(s.left, t) || isSubtree(s.right, t);
    }

    // 判断两棵树是否相同
    private boolean isSameTree(TreeNode p, TreeNode q) {
        if (p == null && q == null) {
            return true;
        }
        if (p == null || q == null) {
            return false;
        }
        if (p.val != q.val) {
            return false;
        }
        // 递归检查左右子树
        return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
    }

    public static void main(String[] args) {
        // 构建树 s
        TreeNode s = new TreeNode(3);
        s.left = new TreeNode(4);
        s.right = new TreeNode(5);
        s.left.left = new TreeNode(1);
        s.left.right = new TreeNode(2);

        // 构建树 t
        TreeNode t = new TreeNode(4);
        t.left = new TreeNode(1);
        t.right = new TreeNode(2);

        SubtreeOfAnotherTree solution = new SubtreeOfAnotherTree();
        boolean result = solution.isSubtree(s, t);
        System.out.println("t 是否是 s 的子树: " + result);
    }
}