剑指 Offer 26. 树的子结构
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
示例 1:
输入: A = [1,2,3], B = [3,1]
输出: false
示例 2:
输入: A = [3,4,5,1,2], B = [4,1]
输出: true
思路:先在A中找到于B根节点相同的节点,再进行比较判断
//判断函数,利用递归实现比较
bool Compare(TreeNode* A, TreeNode* B){
// 若B已经无后续,则表明递归完毕,B为A的子结构
if(B == NULL) return true;
// 若B不为空并且A为空或者A与B的值不相等,直接可以判断B不是A的子结构
if(A == NULL || A->val != B->val) return false;
//该节点相同,则进入下一层比较
return Compare(A->left,B->left)&&Compare(A->right,B->right);
}
这里有个大错误:当A->val==B->val时,直接return 就寄了
//在A中找B节点相同的节点(错误版本)
bool isSubStructure(TreeNode* A, TreeNode* B) {
if(A==NULL||B==NULL) return false;
//如果A B指向的节点相同,则进入判断是否具有相同的结构的函数
if(A->val==B->val) return Compare(A,B);
return isSubStructure(A->left,B)||isSubStructure(A->right,B);
}
//在A中找B节点相同的节点
bool isSubStructure(TreeNode* A, TreeNode* B) {
if(A==NULL||B==NULL) return false;
//有可能后面还有相同的节点入口,所以不能直接return
return Compare(A,B)||isSubStructure(A->left,B)||isSubStructure(A->right,B);
}