Java实现 LeetCode 572 另一个树的子树(遍历树)

572. 另一个树的子树

给定两个非空二叉树 s 和 t，检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。

示例 1:

给定的树 s:

     3
    / \
   4   5
  / \
 1   2

给定的树 t：

   4
  / \
 1   2

返回 true，因为 t 与 s 的一个子树拥有相同的结构和节点值。

示例 2:

给定的树 s：

     3
    / \
   4   5
  / \
 1   2
    /
   0

给定的树 t：

   4
  / \
 1   2

返回 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) {
        return isSame(s, t, false);
    }

    public boolean isSame(TreeNode s, TreeNode t, boolean flag){
        if(s==null && t==null) return true;
        else if(s!=null && t!=null){
            if(s.val == t.val){
                return (isSame(s.left, t.left, true) && isSame(s.right, t.right, true)) || isSame(s.left, t, false) || isSame(s.right, t, false);
            }else{
                if(flag) return false;
                else return isSame(s.left, t, false) || isSame(s.right, t, false);
            }
        }
        return false;
    }
}