1、题目描述
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
限制:
0 <= 节点个数 <= 10000
2、示例
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B:
4
/
1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
输入:A = [1,2,3], B = [3,1]
输出:false
输入:A = [3,4,5,1,2], B = [4,1]
输出:true
3、题解
基本思想:双重递归,第一重递归遍历A每一个节点,如果当前节点值与B的根节点值相等,进入第二重递归,判断两棵树结构和数值是否相等。
/**
* 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 res=false;
bool isSubStructure(TreeNode* A, TreeNode* B) {
if(B==nullptr)
return false;
dfs(A,B);
return res;
}
void dfs(TreeNode* A,TreeNode* B)
{
if(A==nullptr)
return;
if(A->val==B->val)
{
if(Recursion(A,B))
{
res=true;
return;
}
}
dfs(A->left,B);
dfs(A->right,B);
}
bool Recursion(TreeNode* A,TreeNode* B)
{
if(B==nullptr)
return true;
if(A==nullptr)
return false;
if(A->val!=B->val)
return false;
if(Recursion(A->left,B->left)&&Recursion(A->right,B->right))
return true;
return false;
}
};