树的子结构(代码的鲁棒性)

来源:互联网 发布:好看的步兵番号知乎 编辑:程序博客网 时间:2024/06/12 01:43

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

public class TreeNode {    int val = 0;    TreeNode left = null;    TreeNode right = null;    public TreeNode(int val) {        this.val = val;    }}

思路:递归

1、首先设置标志位result = false,一旦匹配成功result就设为true,剩下的代码不会执行,如果匹配不成功,默认返回false。

2、递归思想,如果根节点相同则递归调用DoesTree1HaveTree2(),如果根节点不相同,则判断Tree1的左子树和Tree2是否相同,再判断右子树和Tree2是否相同。

3、注意null的条件,HasSubTree中,如果两棵树都不为空才进行判断。

DoesTree1HaveTree2中,

Tree2为空,则说明第二棵树遍历完了,即匹配成功;

Tree1为空有两种情况:(1)如果tree1为空&&tree2不为空,说明不匹配;(2)如果tree1为空,tree2为空,说明匹配。


public class Solution {    public boolean HasSubtree(TreeNode root1,TreeNode root2) {        boolean result = false;        //当Tree1和Tree2都不为零的时候,才进行比较。否则直接返回false        if (root1 != null && root2 != null)        {            //如果找到了对应Tree2的根节点的点            if (root1.val == root2.val)                //以这个根节点为为起点判断是否包含Tree2                result = DoesTree1HaveTree2(root1, root2);            //如果找不到,那么就再去root的左儿子当作起点,去判断时候包含Tree2            if (!result) result = DoesTree1HaveTree2(root1.left, root2);            //如果还找不到,那么就再去root的右儿子当作起点,去判断时候包含Tree2            if (!result) result = DoesTree1HaveTree2(root1.right, root2);        }        //返回结果        return result;    }    public boolean DoesTree1HaveTree2(TreeNode root1, TreeNode root2)    {        //如果Tree2还没有遍历完,Tree1却遍历完了。返回false        if (root1 == null && root2 != null) return false;        //如果Tree2已经遍历完了都能对应的上,返回true        if (root2 == null) return true;        //如果其中有一个点没有对应上,返回false        if (root1.val != root2.val) return false;        //如果根节点对应的上,那么就分别去子节点里面匹配        return DoesTree1HaveTree2(root1.left, root2.left) && DoesTree1HaveTree2(root1.right, root2.right);    }}


原创粉丝点击