剑指 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
思路:运用递归(自己再实现一个函数递归),如果当前节点与B树的根节点相同,则继续将它们各自的左子树右子树相比较,如果B是该节点的子树,则返回true,否则,将当前节点的左子树、右子树再分别与B比较,直到A树到了叶子节点还没有匹配上,说明不是A树的子树。递归结束条件是:如果B树向下递归到了NULL,返回true,而如果B树没有到NULL,而A树到了NULL,返回false,只要A有一颗子树与B树结构相同,最终结果就返回true
注意:实现函数时,记得穿3个参数,一个是A的子树,一个是B的子树,还有一个是B,因为递归时,会存在一开始匹配而之后不匹配的情况,此时要重新开始比较,而这样B就已经不再是原来的B了,而是B的某个子树,所以要传一个B,在不匹配时,能够再使用最初的B
这里有一个坑,以下图为例
A树:
B树:
写递归条件时不可盲目写成下面这种形式:
//写的太绝对了,一旦根节点匹配而左右子树不匹配,返回就为false了
if (A->val == b->val)
{
return _isSubStructure(A->left, b->left, B) && _isSubStructure(A->right, b->right, B);
}
else
{
return _isSubStructure(A->left, B, B) || _isSubStructure(A->right, B, B);
}
因为一旦最初的根节点暂时匹配上了,而左子树匹配了,右子树没有匹配,因为是&&操作符,一个为假就都为假,所以最终结果就是假
下图是分析,图中的N代表NULL,箭头从上到下是递归向下,箭头从下到上是递归返回
代码:
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
};
bool _isSubStructure(struct TreeNode*A, struct TreeNode*b, struct TreeNode*B)
{
if (b == NULL)
{
return true;
}
if (A == NULL)
return false;
if (A->val == b->val)
{
if (_isSubStructure(A->left, b->left, B) && _isSubStructure(A->right, b->right, B))//如果左右子树都匹配返回true
{
return true;
}
else//不匹配的话就继续向下匹配
{
return _isSubStructure(A->left, B, B) || _isSubStructure(A->right, B, B);
}
}
else
{
return _isSubStructure(A->left, B, B) || _isSubStructure(A->right, B, B);
}
}
bool isSubStructure(struct TreeNode* A, struct TreeNode* B){
if (A == NULL || B == NULL)
return false;
return _isSubStructure(A, B, B);
}