赞
踩

给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。
样例输入:root = [3,4,5,1,2], subRoot = [4,1,2]
样例输出:true
bool isSubtree(struct TreeNode* root, struct TreeNode* subRoot){
}

先实现一个函数isSameTree(struct TreeNode* p, struct TreeNode* q)用于判断p和q是否是同一棵树,然后遍历root,对每个它的子树都和目标subRoot进行isSameTree判定,直到有一棵子树是满足情况的;
由于需要把所有结点都遍历一遍,每个结点遍历都需要进行一次树的比较,所以时间复杂度为 O ( n 2 ) O(n^2) O(n2)。
bool isSameTree(struct TreeNode* p, struct TreeNode* q){
if(p && !q || !p && q) {
return false; // (1)
}
if(!p && !q) {
return true; // (2)
}
if(p->val != q->val) {
return false; // (3)
}
// (4)
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
bool isSubtree(struct TreeNode* root, struct TreeNode* subRoot){
if( isSameTree(root, subRoot) ) {
return true; // (5)
}
if(root == NULL) {
return false; // (6)
}
if( isSubtree(root->left, subRoot) ) {
return true; // (7)
}
if( isSubtree(root->right, subRoot) ) {
return true; // (8)
}
return false;
}
递归时处理二叉树问题很好的手段。

相信看我文章的大多数都是「 大学生 」,能上大学的都是「 精英 」,那么我们自然要「 精益求精 」,如果你还是「 大一 」,那么太好了,你拥有大把时间,当然你可以选择「 刷剧 」,然而,「 学好算法 」,三年后的你自然「 不能同日而语 」。
那么这里,我整理了「 几十个基础算法 」 的分类,点击开启:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。