题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。
我们规定空树不是任何树的子结构。
样例
树A:
8
/ \
8 7
/ \
9 2
/ \
4 7
树B:
8
/ \
9 2
返回 true ,因为B是A的子结构。
分析:
本题考察二叉树的匹配,用类似于字符串匹配一样暴力的方法逐一比较,找到解则返回true。
主体思路是,我们先按先序遍历顺序枚举第一棵树的结点,再判断以该结点为树根的子树与第二棵树是否匹配。
我们知道,hasSubtree(t1,t2)是用来判断t2是否是以t1为根结点树的子结构。那么hasSubtree(t1->left,t2)和hasSubtree(t1->right,t2)则是判断以t1的左右子树是否包含t2这样的子结构,这样问题就递归的转化为了规模更小的子问题。我们需要写一个函数来判断以t1为根结点的树是否包含t2,并且要求这时必须是从t1开始匹配的,而不能不匹配再下传。
对于isSubtree函数,边界情况自然是t2访问到了空节点,说明对这一块的匹配已经完成,如果t1到了空节点或者两个树对应节点不等,则失配。通过了两个边界条件,说明当前结点是匹配的,接下来要左右结点均匹配才整体匹配。
注意两个函数的区别:
1.hasSubtree是题目要求我们实现的函数,只要t2是t1的子结构即可;而isSubtree则是必须从t1的根结点开始匹配的匹配函数。
2.注意这两个递归函数的递归语句,第一个函数是或连接两个递归语句,因为只要在其左右子树找到一个满足题意的子结构即可;而第二个函数是且连接两个递归语句,一棵树与另一棵树从根结点对应匹配,当且仅当每一个结点都匹配。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool hasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) {
if(!pRoot1 || !pRoot2) return false;
if(isSubtree(pRoot1,pRoot2)) return true;
return hasSubtree(pRoot1->left,pRoot2) || hasSubtree(pRoot1->right,pRoot2);
}
bool isSubtree(TreeNode* t1,TreeNode* t2){
if(!t2) return true;
if(!t1 || t1->val != t2->val) return false;
return isSubtree(t1->left,t2->left) && isSubtree(t1->right,t2->right);
}
};