剑指 Offer 26. 树的子结构——树的遍历+判断

剑指 Offer 26. 树的子结构

题目描述

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:

    3
   / \
  4   5
 / \
1   2

给定的树 B：

  4 
 /
1

返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false
示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true

解题思路：

1. 先序遍历树A中的每个节点（对应函数isSubStructure（A,B））

2. 判断树A中的每个节点为根节点的子树是否包含树B。（对应函数isSub（A,B））

recur(A, B) 函数：

终止条件：

当节点 B 为空：说明树 B 已匹配完成（越过叶子节点），因此返回 true ；
当节点 A 为空：说明已经越过树 A 叶子节点，即匹配失败，返回 false ；
当节点 A 和 B 的值不同：说明匹配失败，返回 false ；

返回值：

判断 A 和 B 的左子节点是否相等，即 isSub(A.left, B.left) ；
判断 A 和 B 的右子节点是否相等，即 isSub(A.right, B.right) ；

isSubStructure(A, B) 函数：

特例处理： 当 树 A 为空 或 树 B 为空 时，直接返回 false ；
返回值： 若树 B 是树 A 的子结构，则必满足以下三种情况之一，因此用或 || 连接；

• 以 节点 A 为根节点的子树 包含树 B ，对应 isSub(A, B)；
• 树 B 是 树 A 左子树 的子结构，对应 isSubStructure(A.left, B)；
• 树 B 是 树 A 右子树 的子结构，对应 isSubStructure(A.right, B)；

class Solution{
	public boolean isSubStructure(TreeNode A, TreeNode B){
		// 1.定义一个标志位flag用来做判断
		boolean flag = false;

		if (A != null && B != null) {
			// 2.1 如果从根结点就找到，并且递归成功，返回true
			flag = isSub(A, B);
			// 2.2 如果根结点开始的没有找到，从根结点的左子结点开始搜索
			if (!flag) {
				flag = isSubStructure(A.left, B);
			}
			// 2.3 如果以上都没有找到，就从根结点的右子节点开始搜索
			if (!flag) {
				flag = isSubStructure(A.right,B);
			}
		}

		// 3.最后返回flag即得到结果
		return flag;
	}


	//函数DoesTree1HasTree2()实现判定A树是否包含B树结构的功能
	boolean isSub(TreeNode A, TreeNode B){
		// 1.首先判断B是否为空，一旦B为空，说明B已经递归到叶子结点
		if (B == null) {
			return true;
		}
		// 2.如果A == null，则返回false
		if (A == null) {
			return false;
		}

		// 3.如果A.val != B.val，返回false
		if (A.val != B.val) {
			return false;
		}

		// 4.如果以上情况都不是，说明A.val == B.val，我们继续向下递归判断
		return isSub(A.left, B.left) && isSub(A.right, B.right);

	}
}

内存优化后：

public boolean isSubStructure(TreeNode A, TreeNode B) {
    //边界条件判断，如果A和B有一个为空，返回false
    if (A == null || B == null)
        return false;
    return isSub(A, B);
}

boolean isSub(TreeNode A, TreeNode B) {
    //这里如果B为空，说明B已经访问完了，确定是A的子结构
    if (B == null)
        return true;
    //如果B不为空A为空，或者这两个节点值不同，说明B树不是
    //A的子结构，直接返回false
    if (A == null || A.val != B.val)
        return false;
    //当前节点比较完之后还要继续判断左右子节点
    return isSub(A.left, B.left) && isSub(A.right, B.right);
}

参考链接：https://leetcode-cn.com/u/jyd/