树的子结构

来源:互联网 发布:灰鸽子软件官网 编辑:程序博客网 时间:2024/05/17 05:12

题目:输入两颗二叉树A,B,判断B是不是A的子结构。

有两个任务,第一是比较B树和A的以某个根节点开始一个局部树是否完全相等(isSub()),第二是遍历A(HasSubtree())。代码如下:

public class Solution {    public boolean HasSubtree(TreeNode root1,TreeNode root2) {        if(root1==null || root2==null) return false;        if(root1.val==root2.val && isSub(root1, root2)) return true;        else return (HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2));    }    // Compare only when root1.val == root2.val    private boolean isSub(TreeNode root1, TreeNode root2) {        if(root2.left==null && root2.right==null) return true;        if(root2.left==null) { // root2.right not null            if(root1.right==null) return false;            if(root1.right.val!=root2.right.val) return false;            return isSub(root1.right, root2.right);        }        if(root2.right==null) { // root2.left not null            if(root1.left==null) return false;            if(root1.left.val!=root2.left.val) return false;            return isSub(root1.left, root2.left);        }        else { // root2.left not null and root2.right not null            if(root1.left==null || root1.right==null) return false;            if(root1.left.val!=root2.left.val || root1.right.val!=root2.right.val) return false;            return (isSub(root1.left, root2.left) && isSub(root1.right, root2.right));        }       }}
0 0