题目
检查子树。你有两棵非常大的二叉树:T1,有几万个节点;T2,有几万个节点。设计一个算法,判断 T2 是否为 T1 的子树。
如果 T1 有这么一个节点 n,其子树与 T2 一模一样,则 T2 为 T1 的子树,也就是说,从节点 n 处把树砍断,得到的树与 T2 完全相同。
注意:此题相对书上原题略有改动。
题解 递归判断
单独写一个判断函数,如果结点不同,则返回false
,如果两个树都为空了,说明前面所有结点都相同,返回true
,一树空,一树不空,说明两树存在差异,返回false
。
/**
* 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 Judge(TreeNode* t1, TreeNode* t2){
if(t1 == NULL && t2 == NULL)
return true;
if(t1 == NULL || t2 == NULL)
return false;
if(t1->val != t2->val)
return false;
return Judge(t1->left, t2->left) && Judge(t1->right, t2->right);
}
bool checkSubTree(TreeNode* t1, TreeNode* t2) {
if(t1 == NULL && t2 == NULL)
return true;
if(t1 == NULL || t2 == NULL)
return false;
return Judge(t1, t2) || checkSubTree(t1->left, t2) ||checkSubTree(t1->right, t2);
}
};