剑指offer:(18)代码的鲁棒性:树的子结构

35 篇文章 0 订阅
22 篇文章 0 订阅

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

测试用例:

       功能测试:

             树1和树2都是普通的二叉树,树2是或者不是树1的子结构

        特殊输入测试:

              两颗二叉树的一个或者两个根节点为null、

              二叉树的所有节点都没有左子树或者右子树

方法一:

       思路:参考剑指offer 

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

            2、递归思想,如果根节点相同则递归调用DoesTree1HaveTree2(),  如果根节点不相同,则判断tree1的左子树和tree2是否相同, 再判断右子树和tree2是否相同
 3、注意null的条件,HasSubTree中,如果两棵树都不为空才进行判断,DoesTree1HasTree2中,如果Tree2为空,则说明第二棵树遍历完了,即匹配成功,tree1为空有两种情况

                        (1)如果tree1为空&&tree2不为空说明不匹配, 

                        (2)如果tree1为空,tree2为空,说明匹配。

package co.com.jianzhioffer;

public class Solution18 {
	public class TreeNode {
		int val;
		TreeNode left = null;
		TreeNode right = null;

		TreeNode(int val) {
			this.val = val;
		}

	}

	
   /*	 
     * 递归实现,方法一
	 * 思路:参考剑指offer 1、首先设置标志位result = false,因为一旦匹配成功result就设为true,
	 * 剩下的代码不会执行,如果匹配不成功,默认返回false 2、递归思想,如果根节点相同则递归调用DoesTree1HaveTree2(),
	 * 如果根节点不相同,则判断tree1的左子树和tree2是否相同, 再判断右子树和tree2是否相同
	 * 3、注意null的条件,HasSubTree中,如果两棵树都不为空才进行判断,
	 * DoesTree1HasTree2中,如果Tree2为空,则说明第二棵树遍历完了,即匹配成功,
	 * tree1为空有两种情况(1)如果tree1为空&&tree2不为空说明不匹配, (2)如果tree1为空,tree2为空,说明匹配。
     */
	 
	public boolean HasSubtree(TreeNode root1, TreeNode root2) {
		boolean result = false;
		if (root1 != null && root2 != null) {
			if (root1.val == root2.val)
				result = DoesTree1HaveTree2(root1, root2);
			if (!result)
				result = HasSubtree(root1.left, root2);
			if (!result)
				result = HasSubtree(root1.right, root2);

		}
		return result;
	}

	public boolean DoesTree1HaveTree2(TreeNode root1, TreeNode root2) {
		if (root2 == null) 
			return true;
		if (root1 == null)
			return false;

		if (root1.val != root2.val) 
			return false;

		return DoesTree1HaveTree2(root1.left, root2.left)
				&& DoesTree1HaveTree2(root1.right, root2.right);
	}
}

方法二:上面方法的简洁版,利用好短路特性,不需要那么多flag

public boolean HasSubtree(TreeNode root1, TreeNode root2) {
		if(root1 ==null || root2 == null) return false;
		return isSubtree(root1, root2) || HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2);
	}
	public boolean isSubtree(TreeNode root1, TreeNode root2){
		if(root2 == null) return true;
		if(root1 == null) return false;
		if(root1.val == root2.val)
			return isSubtree(root1.left, root2.left) && isSubtree(root1.right, root2.right);
		else return false;
	}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值