输入两棵二叉树A,B,判断B是不是A的子结构。(我们约定空树不是任意一个树的子结构)
假如给定A为{8,8,7,9,2,#,#,#,#,4,7},B为{8,9,2},2个树的结构如下,可以看出B是A的子结构
代码:
#define FALSE 0
#define TRUE 1
bool has_sub(struct TreeNode* pRoot1, struct TreeNode* pRoot2){
if(pRoot1 == NULL && pRoot2 != NULL){
return FALSE;
}
if(pRoot2 == NULL){
return TRUE;
}
if(pRoot1->val != pRoot2->val){
return FALSE;
}
return has_sub(pRoot1->left, pRoot2->left) && has_sub(pRoot1->right, pRoot2->right);
}
bool pre_order(struct TreeNode* pRoot1, struct TreeNode* pRoot2) {
if(pRoot1 == NULL){
return FALSE;
}
if(pRoot1->val == pRoot2->val){
if(has_sub(pRoot1, pRoot2)){
return TRUE;
}
}
return pre_order(pRoot1->left, pRoot2) || pre_order(pRoot1->right, pRoot2);
}
bool HasSubtree(struct TreeNode* pRoot1, struct TreeNode* pRoot2 ) {
if(pRoot1 == NULL || pRoot2 == NULL){
return FALSE;
}
return pre_order(pRoot1, pRoot2);
}
一个复杂的递归问题。
首先要思考,如何判断是不是子树。首先根节点值相同,而且后续的每个节点都要相同。
我们对主树进行前序遍历(根左右),首先判断根节点是不是等于子树的根节点,如果是,那接着判断剩余节点;如果不等于,那就判断主树的子节点是不是等于子树的根节点。它们之间的关系为或,即只要找到一个证明是子树,那就可以返回true。
对于判断其余节点是不是相同同样适用前序遍历,当主树为NULL,而子树有值时,说明不是子树;当值不相等时也为false,当所有之前的值都相等,而且子树为NULL时,则说明是子树。要求所有节点都相等所以是与的逻辑关系。