[LeetCode] 572. Subtree of Another Tree

原题链接： https://leetcode.com/problems/subtree-of-another-tree/

1. 题目介绍

Given two non-empty binary trees s and t, check whether tree t has exactly the same structure and node values with a subtree of s. A subtree of s is a tree consists of a node in s and all of this node’s descendants. The tree s could also be considered as a subtree of itself.
给出两个非空的二叉树 s 和 t 。判断 t 是否是 s 的一个子树。 s 的子树是指 s 的某个节点及其所有后代共同组成的一棵树。s的子树也包括s本身。

Example 1:
Given tree s:

Given tree t:

Return true, because t has the same structure and node values with a subtree of s.
返回true，因为 t 有和s的子树相同的结构和节点

Example 2:
Given tree s:

Given tree t:

Return false.
返回false

2. 解题思路

首先写一个判断两个树是否相等的函数 isSame，该函数用递归判断两数是否相等。
然后再用深度优先搜索的方式遍历整个 s 树，如果遇到和 t 树根节点 val 相同的节点，就用 isSame 函数比较一下是否和 t 树相等。如果相等，直接返回 true，如果不相等，继续遍历 s 树。

实现代码

/**
 * 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) {
        if(s == null){
            return false;
        }
        if(s.val == t.val && isSame(s,t) == true){
            return true;
        }
        return isSubtree(s.left,t) || isSubtree(s.right,t);
    }
    
    //isSame函数判断两个树是否相等
    public boolean isSame(TreeNode t1,TreeNode t2){
        if(t1 == null && t2 == null){
            return true;
        }
        if(t1 == null || t2 == null){
            return false;
        }
        if(t1.val != t2.val){
            return false;
        }
        return isSame(t1.left,t2.left) && isSame(t1.right,t2.right);
    }   
}

如果将上述代码的 isSubtree(TreeNode s, TreeNode t) 函数稍作改进，先判断 s 的子树是否和 t 相等，再判断 s 自己是否和 t 相等，这样就不必比较s.val 和 t.val 是否相等了，执行的速度会有很大的提高。同时也可以使用三目运算符将其写在同一行内，使得代码看上去更加简洁。

实现代码

/**
 * 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) {
        if(s == null) return false;
        return (isSubtree(s.left, t) || isSubtree(s.right, t) ) ? true : isSame(s, t);
    }
    //isSame函数判断两个树是否相等
    public boolean isSame(TreeNode t1,TreeNode t2){
        if(t1 == null && t2 == null){
            return true;
        }
        if(t1 == null || t2 == null){
            return false;
        }
        if(t1.val != t2.val){
            return false;
        }
        return isSame(t1.left,t2.left) && isSame(t1.right,t2.right);
    }   
}