LeetCode算法题-Subtree of Another Tree（Java实现）

Posted 2021-02-11 程序员小川

这是悦乐书的第265次更新，第278篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第132题（顺位题号是572）。给定两个非空的二进制树s和t，检查树t是否具有完全相同的结构和具有子树s的节点值。 s的子树是一个树，由s中的节点和所有节点的后代组成。 树也可以被视为自己的子树。例如：

鉴于树s：

      3
     /     4   5
   /   1   2

鉴于树t：

   4 
  /  1   2

返回true，因为t具有相同的结构和节点值，其子树为s。


鉴于树s：

     3
    /    4   5
  /  1   2
    /
   0

鉴于树t：

   4 
  /  1   2

返回false。

本次解题使用的开发工具是eclipse，jdk使用的版本是1.8，环境是win7 64位系统，使用Java语言编写和测试。

02 第一种解法

要判断t是不是s的子树，最直接的就是遍历s的节点，以每一个s的节点为新的子树去和t比较，而比较的方法需要借助一个递归的方法来完成，来判断两个二叉树是不是相同的结构和相同的值。所以我们需要先使用一个队列，来遍历s的每一个节点，拿到每一个节点后，作为新的二叉树去和t比较，直到找到合适节点开始的子树与t相同，否则返回false。

public boolean isSubtree(TreeNode s, TreeNode t) {
    Queue<TreeNode> queue = new LinkedList<TreeNode>();
    queue.offer(s);
    while (!queue.isEmpty()) {
        TreeNode node = queue.poll();
        if (node.left != null) {
            queue.offer(node.left);
        }
        if (node.right != null) {
            queue.offer(node.right);
        }
        if (isSame(node, t)) {
            return true;
        }
    }
    return false;
}

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

03 第二种解法

我们也可以把第一种解法的队列去掉，也换成递归的方法，但是思路还是一样的，需要拿s的每一个节点去比对。

public boolean isSubtree2(TreeNode s, TreeNode t) {
    if (s == null) {
        return false;
    }
    return isSame2(s, t) || isSubtree2(s.left, t) || isSubtree2(s.right, t); 
}

public boolean isSame2(TreeNode s, TreeNode t){
    if (s == null  ||  t == null) {
        return s == t;
    }
    return s.val == t.val && isSame(s.left, t.left) && isSame(s.right, t.right);
}

04 第三种解法

将上面的解法再精简下，只使用一个递归方法。我们知道，只有在s中找到合适的开始节点值（与t的根节点值相等），才能去判断以该节点作为开始节点的子树是否和t是相同结构的，找不到就只能继续找s的左节点值或右节点值了。

要想只使用一个递归，那么必须再引入一个参数，来判断当前s的节点值是否与t的根节点值相等。如果相等，那么依次遍历s当前节点的左右节点与t的左右子节点，要是所有节点值都相等，那么就可以返回true。如果不相等，那么就需要兵分两路，s当前节点的左节点与t继续比较，s当前节点的右节点与t继续比较，并且第三个参数为false。

public boolean isSubtree3(TreeNode s, TreeNode t) {
    return isSameTree(s, t, false);
}

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

05 第四种解法

我们也可以借助字符串和其子串的思路，将s和t的每一个节点都遍历出来，组成一个字符串，然后去判断t所组成的字符串是不是s所组成字符串的子串。

public boolean isSubtree4(TreeNode s, TreeNode t) {
    String tree = getTreeString(s, true);
    String tree2 = getTreeString(t, true);
    return tree.indexOf(tree2) > -1;
}

public String getTreeString(TreeNode t, boolean left) {
    if (t == null) {
        return left ? "lnull" : "rnull";
    }
    return "#"+t.val + " " +getTreeString(t.left, true)+" " +getTreeString(t.right, false);
}

06 小结

算法专题目前已日更超过四个月，算法题文章132+篇，公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词，获取系列文章合集。

以上就是全部内容，如果大家有什么好的解法思路、建议或者其他问题，可以下方留言交流，点赞、留言、转发就是对我最大的回报和支持！