剑指 Offer 26. 树的子结构
思路:两层递归
1、递归调用isSubStructure(A, B),先序遍历二叉树A;
2、设置函数traverse,调用函数traverse( A, B),判断二叉树A中是否包含子树B。
令二叉树A的根节点为节点A,二叉树B的根节点为B。
若二叉树A为空或二叉树 B为空 时,直接返回 false;
isSubStructure函数,返回值有三种情况:
1、以节点 A 为根节点的子树包含树 B ,调用 traverse(A, B);
2、当子树 B是 二叉树A的左子树的子结构,调用 isSubStructure(A.left, B);
3、当子树B是 二叉树A右子 的子结构,调用 isSubStructure(A.right, B);
traverse函数:
1、当节点B为空时,越过叶子节点则说明树 B已匹配完成,因此返回 true;
2、当节点 A为空时,说明已经越过二叉树A的叶子节点,则匹配失败,返回 false ;
3、返回值:
(1)判断节点 A和 B的值是否相同;
(2)判断 A 和 B 的左子节点是否相等,即 traverse(A.left, B.left) ;
(3)判断 A 和 B 的右子节点是否相等,即 traverse(A.right, B.right)
class Solution {
public boolean isSubStructure(TreeNode A, TreeNode B) {
if (A == null || B == null) {
return false;
}
return traverse(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
}
private boolean traverse(TreeNode A, TreeNode B) {
if (B == null) {
return true;
} else if (A == null) {
return false;
}
return A.val == B.val && traverse(A.left, B.left) && traverse(A.right, B.right);
}
}