面试题 04.10. 检查子树-----两种解法

题目表述

检查子树。你有两棵非常大的二叉树：T1，有几万个节点；T2，有几万个节点。设计一个算法，判断 T2 是否为 T1 的子树。

如果 T1 有这么一个节点 n，其子树与 T2 一模一样，则 T2 为 T1 的子树，也就是说，从节点 n 处把树砍断，得到的树与 T2 完全相同。

示例：

输入：t1 = [1, 2, 3], t2 = [2]
输出： true

方法1

分别遍历t1 和 t2 两棵树，生成两个字符串，只需要判断t2字符串是否为t1字符串的子串即可。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean checkSubTree(TreeNode t1, TreeNode t2) {
        String sub = dfs(t2);
        String str = dfs(t1);


        return str.contains(sub);
    }
    public String dfs(TreeNode root){
        if(root == null) return "";
        return "" + root.val + dfs(root.left) + dfs(root.right);
    }
}

解法2

递归搜索，找到t1 和 t2 相同的节点，如果相同则分别递归左节点和右节点判断是否相同。如果不相同做递归t1 左子结点和右子节点，找到与t2相同的节点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean checkSubTree(TreeNode t1, TreeNode t2) {
        if(t1 != null && t2 == null) return true;
        if(t1 == null && t2 != null) return false;
        if(t1 == null && t2==null) return true;
        if(t1.val == t2.val){
            return checkSubTree(t1.left, t2.left) && checkSubTree(t1.right, t2.right);
        }else{
            return checkSubTree(t1.left, t2) || checkSubTree(t1.right, t2);
        }
    }


}