匹配二叉树的子结构
问题概述 与 思路剖析
问题概述
匹配二叉树的子结构,(子结构为空树也视为不匹配)
如下图所示:
简单的说,我们将 二叉树与 子结构树 形象化后,子结构树 可以很清楚的看成 二叉树的一部分,且完全匹配(值相等,链接也相同)
思路剖析
由于是二叉树匹配 子结构,我们需要第一步在二叉树中找到第一个符合 子结构的第一个节点的节点,之后进行逐一遍历判断是否完全匹配。
步骤解析:
- 在二叉树中找到匹配 子结构树头节点的第一个节点
- 从上一步找到的节点开始逐个匹配
- 完全匹配的话则返回 true,不完全匹配继续向下遍历,重复 1,2步骤
编写高质量的代码 与 手撕代码展示
编写高质量代码的注意点:
规范性: 我们在手撕(白板或者白纸上)代码时需要注意规范性,尽量清晰写清每一个字母,通过缩进和对齐括号让代码布局合理,同时合理命名代码中的变量和函数。
完整性: 在编码前全面考虑好所有可能的输入,确保写出的代码在完成基本的功能之外,还考虑了边界条件,并做好的错误处理。只有全面考虑好这 3 个方面的代码才是完整的代码。
鲁棒性: 确保自己的程序不会轻易崩溃。平时写代码时,养成防御性编程的习惯,在函数入口判断输入是否有效,对各种无效输入做好处理。
手撕代码
/**
* 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 isSubStructure(TreeNode* A, TreeNode* B) { //在二叉树中找到匹配 子结构树头节点的第一个节点
bool result = false;
if(A != nullptr && B != nullptr){
if(A->val == B->val)
result = DoesTree1HaveTree2(A,B);
if(!result)
result = isSubStructure(A->left,B);
if(!result)
result = isSubStructure(A->right,B);
}
return result;
}
bool DoesTree1HaveTree2(TreeNode* pRoot1,TreeNode* pRoot2){ //判断 1 树中是否含有 2 树
if(pRoot2 == nullptr)
return true;
if(pRoot1 == nullptr)
return false;
if(pRoot1->val != pRoot2->val)
return false;
return DoesTree1HaveTree2(pRoot1->left,pRoot2->left) && DoesTree1HaveTree2(pRoot1->right,pRoot2->right);
}
};