牛客_树的子结构

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhangvalue/article/details/87899829

题目描述

输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

思路:


1、首先设置标志位flag= false,因为一旦匹配成功flag就设为true,
剩下的代码不会执行,如果匹配不成功,默认返回false
2、递归思想,如果根节点相同则递归调用DoesTree1HaveTree2(),
如果根节点不相同,则判断tree1的左子树和tree2是否相同,
再判断右子树和tree2是否相同
3、注意null的条件,HasSubTree中,如果两棵树都不为空才进行判断,
DoesTree1HasTree2中,如果Tree2为空,则说明第二棵树遍历完了,即匹配成功,
tree1为空有两种情况(1)如果tree1为空&&tree2不为空说明不匹配,
(2)如果tree1为空,tree2为空,说明匹配。

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
         boolean flag = false;//作为标记查看是否root2是root1的子树
        //root1和root2都不能为null
        if(root1!=null&&root2!=null) { 
          if(root1.val==root2.val){ flag=DoesTree1HaveTree2(root1,root2);}
          if(!flag){ flag=HasSubtree(root1.left,root2);}
          if(!flag){ flag=HasSubtree(root1.right,root2); }
        }
        return flag;
    }
        boolean DoesTree1HaveTree2(TreeNode root1,TreeNode root2){
            //root2为空说明了,此时被匹配的二叉树为空,已经遍历完了root2,匹配成功
            if(root2==null){return true;}
            //此时说明root2还不为空而root1已经匹配完了,匹配失败
            if(root1==null){return false;}
            //此时root2被匹配的二叉树不为空,root1不为空,再看两个二叉树的val相同,才能继续寻找
            if(root1.val!=root2.val){return false;}
            //此时说明了,root2被匹配的二叉树不为空,root1不为空,且root1和root2的val相同
            return  DoesTree1HaveTree2( root1.left, root2.left)&& DoesTree1HaveTree2( root1.right, root2.right);
        }
}


猜你喜欢

转载自blog.csdn.net/zhangvalue/article/details/87899829