官方题解有很多高大上的方法,我这就将一个最容易想到、最直接的方法吧,比较详细(基本上没有压缩代码),有不懂的可以在评论区问我~~~若有不足欢迎大佬斧正(>人<;)
首先让我们来看看这个二叉树的题目(如下图)
看完题目,明确了一棵树的子树是到叶子节点结束的,这里抛出我开始做题的想法
- 如何找到子树的起始点?
- 找到起始节点后,只能说明起始节点相同,万一树 s 里的子树结构里面有很多与树 t 相似的,仅是里面有一两个值不同,该如何解决?
- 需不需要特别注意 “单边树” 的问题?
顺着这个思路我就写了如下代码,写的过程中经过调试解决了问题,总的可以概括为两个步骤:
- find 树 t 的起始节点
- check 整棵树,不相等再继续往下找(有点像DFS)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
bool check(TreeNode* p, TreeNode* q)
{
//递归结束条件
if(!p && !q) return true;
else if(!p && q || p && !q || p->val != q->val) return false;
bool L = check(p->left, q->left);
bool R = check(p->right, q->right);
return L&R; //整棵树必须相同才行,相当于 if(!L || !R) return false; return true;
}
bool find(TreeNode* s, TreeNode* t)
{
//递归结束条件
if(!s) return false;
bool flag = false; //用来判断找到的树 s 的子树是否全部与 树 t 相等
if(s->val == t->val) flag = check(s, t);
if(flag == true) return flag;
bool L = find(s->left, t);
bool R = find(s->right, t);
return L|R; //只要找到了一个就为真,相当于if(L || R) return true; return false;
}
bool isSubtree(TreeNode* s, TreeNode* t) {
return find(s, t);
}
};
该代码时间复杂度在用时26秒左右。
路漫漫其修远兮,吾将上下而求索