leetcode-剑指 Offer 26. 树的子结构
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
示例 1:
输入:A = [1,2,3], B = [3,1]
输出:false
示例 2:
输入:A = [3,4,5,1,2], B = [4,1]
输出:true
限制:
0 <= 节点个数 <= 10000
解题思路:
由于要判断B是不是A的子结构,那么子结构不同于子树。所以需要进行两个部分的工作:
1、要在A中找到和B根节点相同的节点,如果A中的根节点不同,那么就遍历A的左右子树(此时的函数,只是在A中搜索B的根节点,B始终保持不变)
2、如果在A中找到了B的根节点,那么接下来要做的就是:判断以B的根节点的A这棵子树中是否包含B的结构。
class Solution {
bool dfs(TreeNode* A,TreeNode* B) //判断A中是否包含B结构
{
if(!B) return true;
if(!A) return false;
return A->val==B->val&&dfs(A->left,B->left)&&dfs(A->right,B->right);
}
public:
bool isSubStructure(TreeNode* A, TreeNode* B) {//找A中B的根节点
if(!A||!B) return false;
if(dfs(A,B)) return true;
return isSubStructure(A->left,B)||isSubStructure(A->right,B);
//如果dfs(A,B)==false,证明一开始的A根节点和B的根节点不一样,所以继续去A的左右子树中区找和B一样的根,
//所以dfs中的A每次都是变化的
}
};