剑指 Offer 26. 树的子结构
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:3 / \ 4 5 / \ 1 2
给定的树 B:
4 / 1
示例 1:
输入:A = [1,2,3], B = [3,1]
输出:false
示例 2:
输入:A = [3,4,5,1,2], B = [4,1]
输出:true
限制:
0 <= 节点个数 <= 10000
思路:先序遍历 |
class Solution {
public:
bool isSubStructure(TreeNode* A, TreeNode* B) {
if(!A or !B) return false;
return isOK(A,B) or isSubStructure(A->left,B) or isSubStructure(A->right,B);
}
bool isOK(TreeNode* p, TreeNode* q){
if(!q) return true;
if(!p or p->val != q->val) return false;
return isOK(p->left,q->left) and isOK(p->right,q->right);
}
};