输入两棵二叉树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
题目解析
题目要求查找树 A 中是否存在和树 B 结构一样的子树,只需完成以下两个工作即可:
在树 A 中找到和树 B 的根节点的值一样的节点 root
判断树 A 中以 root 为根节点的子树是不是包含和树 B 一样的结构
算法流程
工作一: 对二叉树进行遍历找到树 A 的根节点和树 B 的根节点相等的位置 root
工作二: 判断树 A 中以 root 为根节点的子树和树 B 具有相同的结构
处理 recur(A, B) 函数
- 终止条件
当节点 B 为空:说明树 B 已完成匹配,返回 true
当节点 A 为空:说明树 A 全部遍历完也无法匹配树 B,返回 false
当节点 A 和 B 的值不等:匹配失败,返回 false
- 返回值
判断 A 和 B 的左子节点是否相等,即 recur(A->left, B->left);
判断 A 和 B 的右子节点是否相等,即 recur(A->right, B->right);
重点思想:将问题拆解
1.找到两个二叉树相等的节点
2.再匹配下面的结构是否相等
在isSubStructure中,通过递归遍历A的每一个节点是否和B的第一个节点相等,如果都不相等,则最终返回false
如果A有一个节点和B相等,则在recur中一起对A和B的左右分支进行比较,中途有一个不相等,则B就不是A的子结构,如果遍历到B的末端都没出现不相等的,则B就是A 的子结构,返回true
class Solution {
public:
bool isSubStructure(TreeNode* A, TreeNode* B) {
// 特判:树 A 为空 或 树 B 为空 时,直接返回 false
if (A == nullptr || B == nullptr) return false;
/*
1. 判断 B 是 A 的子树,recur(A, B);
2. 判断 B 是 A 的左子树,isSubisSubStructure(A->left, B);
3. 判断 B 是 A 的右子树,isSubisSubStructure(A->right, B);
*/
return recur(A, B) || isSubStructure(A->left, B) || isSubStructure(A->right, B);
}
bool recur(TreeNode* A, TreeNode* B) {
if (B == nullptr) return true;
if (A == nullptr || A->val != B->val) return false;
return recur(A->left, B->left) && recur(A->right, B->right);
}
};