输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B:
4
/
1
如果树B是树A的子结构,则子结构的根节点可能为树A的任意一个节点。因此判断树B是否是树A的子结构,需要:
1)先序遍历树A中的每个节点
2)判断树A中以
n
A
n_A
nA为根节点的子树是否包含树B
class Solution {
private:
bool doesTreeAHaveTreeB(TreeNode* A, TreeNode* B) {
if(B == nullptr) return true;
if(A == nullptr) return false;
if(A->val != B->val) return false;
return doesTreeAHaveTreeB(A->left,B->left) && doesTreeAHaveTreeB(A->right,B->right);
}
public:
bool isSubStructure(TreeNode* A, TreeNode* B) {
bool res = false;
if(A != NULL && B != NULL) {
if(A->val == B->val) res = doesTreeAHaveTreeB(A,B);
if(!res) res = isSubStructure(A->left,B);
if(!res) res = isSubStructure(A->right,B);
}
return res;
}
};